search

cpetig / tflite_micro_compiler

generate tflite micro code which bypasses the interpreter (directly calls into kernels)
Apache License 2.0
77 stars 26 forks source link

MicroInterpreter::input(N)/output(N) like interface #9

Closed cpetig closed 4 years ago

cpetig commented 4 years ago

(My) TfLiteMicro code typically uses interpreter->input(0) and interpreter->output(0) to access data and compare dimensions. Also the example uses it https://github.com/tensorflow/tensorflow/blob/edbe5e189c1ec14d3a3386aa29e6118d807d9379/tensorflow/lite/micro/examples/hello_world/main_functions.cc#L82 Implement a similar interface for compiled code returning the TfLiteTensor pointer - this way existing code can be more easily adapted to using compiled code.

cpetig commented 4 years ago

Fixed in https://github.com/cpetig/tflite_micro_compiler/commit/6513f59504942d38adddfffd40c478ba5b2a161a