Support for DECODE operator #3132
Conversation
@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 tensorflow#3131
…em). Cleanup unit test.
|
|
||
| namespace tflite { | ||
|
|
||
| struct DecodeState { |
There was a problem hiding this comment.
Please change this to class per https://google.github.io/styleguide/cppguide.html#Structs_vs._Classes
|
|
||
| namespace tflite { | ||
|
|
||
| struct DecodeStateLUT : public DecodeState { |
There was a problem hiding this comment.
Please rename this to DecodeStateLut per https://google.github.io/styleguide/cppguide.html#General_Naming_Rules
| #include "tensorflow/lite/micro/micro_profiler_interface.h" | ||
|
|
||
| namespace tflite { | ||
|
|
There was a problem hiding this comment.
What do you think about converting DecodeState into an interface called IDecodeAlgorithm? This would clarify its role, and we could then reorganize the helper functions into a separate container class.
| @@ -0,0 +1,344 @@ | |||
| /* Copyright 2025 The TensorFlow Authors. All Rights Reserved. | |||
There was a problem hiding this comment.
Would it be clearer if we split this into separate tests, with each one focused on a specific algorithm?
| }; | ||
|
|
||
| // Align the tensor data the same as a Buffer in the TfLite schema | ||
| alignas(16) const uint8_t kEncodedLUT[] = {0x1B, 0xE4}; |
There was a problem hiding this comment.
It'd be useful to have a comment as to how this is constructed.
| count -= 1; | ||
| } | ||
|
|
||
| // process elements in current channel in groups of 16 |
There was a problem hiding this comment.
Out of curiosity, why 16 instead of 8?
| @@ -0,0 +1,344 @@ | |||
| /* Copyright 2025 The TensorFlow Authors. All Rights Reserved. | |||
|
|
|||
| Licensed under the Apache License, Version 2.0 (the "License"); | |||
There was a problem hiding this comment.
LUT has many combinations to cover. Then, do you have a plan to expand the coverage a bit?
3 participants
@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