mirror of
https://github.com/vee1e/tflite-micro.git
synced 2026-09-01 17:57:27 +00:00
a0f8970856
As the next step in the codegen experiment, we want to generate the invoke calls for each layer. This is slightly challenging with the existing sources, as kernels only expose a registration function, not their individual Eval functions. In an effort to keep the code churn to a minimum, this PR introduces an inference only registration structure and function. It includes just two function pointers: invoke and reset. For this CL, we've only introduced it for FullyConnected. In the code generator, this PR creates a new op_table array in the generated source, with an enum for lookup. It also generates an invoke function for each subgraph, that calls each operator's invoke function. BUG=295174388 |
||
|---|---|---|
| .. | ||
| fully_connected.cc | ||
| fully_connected_int8.cc | ||
| hexagon_fully_connected.h | ||
| hexagon_svdf.h | ||
| hexagon_tflm_translation_fully_connected.h | ||
| hexagon_tflm_translation_svdf.h | ||
| LICENSE | ||
| svdf.cc | ||
| svdf_int8.cc | ||