vee1e/tflite-micro
1
0
Fork 0
mirror of https://github.com/vee1e/tflite-micro.git synced 2026-09-02 18:27:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
tflite-micro/codegen/templates
History
Exact
RJ Ascani 3323a41d69
Implement MicroContext/Graph for codegen (#2218) 
The TFLM kernels use the MicroContext and MicroGraph interfaces to fetch
eval tensors and access other parts of the graph. Since codegen does not
have the MicroInterpreter and related objects to serve those, this PR
introduces a new MicroCodegenContext class that serves as both the
MicroContext and MicroGraph.

The MicroCodegenContext is configured with a span of Subgraphs, each of
which includes the inputs, outputs, nodes, tensors and an invocation
function. The code generator will create the data and functions needed
for each Subgraph and initialize the MicroCodegenContext with it. By
having the re-usable MicroCodegenContext, the code generator won't have
to generate nearly as much code.

BUG=b/295174086
2023-09-12 22:28:30 +00:00
..
inference.cc.mako Implement MicroContext/Graph for codegen (#2218) 2023-09-12 22:28:30 +00:00
inference.h.mako Implement MicroContext/Graph for codegen (#2218) 2023-09-12 22:28:30 +00:00