Skip to content

Commit 6264a69

Browse files
ddavis-2015veblush
andauthored
Support for DECODE operator (#3239)
* Support for DECODE operator @tensorflow/micro Add initial support for DECODE operator. Add reference implementation. Add LUT decompression support. Update op resolvers. Update Makefiles and Bazel BUILD files. Add kernel unit test. bug=fixes #3131 * update copyright * Don't use constructors with global objects (bluepill will not call them). Cleanup unit test. * Support for DECODE operator @tensorflow/micro Additional support for DECODE operator. Add Xtensa optimizations for LUT decompression. Move all Xtensa kernel source references to the Xtensa target makefile. bug=fixes #3150 * Updates to Xtensa makefiles @tensorflow/micro Reorganize Xtensa makefiles such that all references to optimized kernel sources are moved to the Xtensa target makefile. Move hifimini kernel sources to the parent directory, and rename them so they do not interfere with the target overlay mechanism of the root makefile. bug=fixes #3153 * Fix incorrect include path. Fix code style errors. * fix copyright * update generic benchmark op resolver size * Support for DECODE operator @tensorflow/micro Add reference implementation of pruning to DECODE operator. Makefile and Bazel BUILD file changes. Additional unit tests. bug=fixes #3161 * Split decode tests into seperate files. Update pruning code with zero-point checks. Add const-tensor checks. * Add decode_test_helpers.h file. Cleanup tests. * Support for DECODE operator @tensorflow/micro Add comprehensive DECODE unit test for LUT decompression. Update decode_test_helpers.h with additional expected comparison tensor types. Update Makefiles and Bazel build files. * post review changes --------- Co-authored-by: Esun Kim <veblush@google.com>
1 parent 3ea59f5 commit 6264a69

File tree

4 files changed

+497
-0
lines changed

4 files changed

+497
-0
lines changed

tensorflow/lite/micro/kernels/BUILD

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -664,6 +664,22 @@ tflm_cc_test(
664664
],
665665
)
666666

667+
tflm_cc_test(
668+
name = "decode_state_lut_test",
669+
srcs = [
670+
"decode_state_lut_test.cc",
671+
],
672+
deps = [
673+
":decode_test_helpers",
674+
":kernel_runner",
675+
"//tensorflow/lite/c:common",
676+
"//tensorflow/lite/micro:debug_log",
677+
"//tensorflow/lite/micro:op_resolvers",
678+
"//tensorflow/lite/micro:test_helpers",
679+
"//tensorflow/lite/micro/testing:micro_test",
680+
],
681+
)
682+
667683
tflm_cc_test(
668684
name = "decode_state_prune_test",
669685
srcs = [

tensorflow/lite/micro/kernels/Makefile.inc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -123,6 +123,7 @@ $(TENSORFLOW_ROOT)tensorflow/lite/micro/kernels/ceil_test.cc \
123123
$(TENSORFLOW_ROOT)tensorflow/lite/micro/kernels/comparisons_test.cc \
124124
$(TENSORFLOW_ROOT)tensorflow/lite/micro/kernels/concatenation_test.cc \
125125
$(TENSORFLOW_ROOT)tensorflow/lite/micro/kernels/cumsum_test.cc \
126+
$(TENSORFLOW_ROOT)tensorflow/lite/micro/kernels/decode_state_lut_test.cc \
126127
$(TENSORFLOW_ROOT)tensorflow/lite/micro/kernels/decode_state_prune_test.cc \
127128
$(TENSORFLOW_ROOT)tensorflow/lite/micro/kernels/decode_test.cc \
128129
$(TENSORFLOW_ROOT)tensorflow/lite/micro/kernels/depth_to_space_test.cc \

0 commit comments

Comments
 (0)