ONE - On-device Neural Engine
|
Loads TF lite file and provides helpers to access attributes. More...
#include <Reader.h>
Public Member Functions | |
Reader (const tflite::Model *model) | |
Reader ()=delete | |
const std::vector< const tflite::OperatorCode * > & | opcodes () |
const TFliteBuffers_t * | buffers () |
const TFliteTensors_t * | tensors () |
const TFliteOperators_t * | operators () |
const std::vector< int32_t > & | inputs () const |
const std::vector< int32_t > & | outputs () const |
uint32_t | num_subgraph () const |
size_t | buffer_info (uint32_t buf_idx, const uint8_t **buff_data) |
tflite::BuiltinOperator | builtin_code (const tflite::Operator *op) const |
std::string | opcode_name (const tflite::Operator *op) const |
bool | select_subgraph (uint32_t subgraph) |
Loads TF lite file and provides helpers to access attributes.
tflinspect::Reader::Reader | ( | const tflite::Model * | model | ) |
Definition at line 28 of file Reader.cpp.
References opcodes().
|
delete |
size_t tflinspect::Reader::buffer_info | ( | uint32_t | buf_idx, |
const uint8_t ** | buff_data | ||
) |
Definition at line 40 of file Reader.cpp.
References size.
|
inline |
tflite::BuiltinOperator tflinspect::Reader::builtin_code | ( | const tflite::Operator * | op | ) | const |
Definition at line 68 of file Reader.cpp.
References mio::tflite::builtin_code_neutral().
Referenced by tflinspect::DumpConv2DWeight::run().
|
inline |
Definition at line 60 of file Reader.h.
Referenced by validate_onnx2circle.OnnxRunner::feed_random_inputs(), and package.infer.session::set_inputs().
|
inline |
Definition at line 63 of file Reader.h.
References flatbuffers::Vector< T >::size().
Referenced by tflinspect::DumpOperators::run(), tflinspect::DumpConv2DWeight::run(), and tflinspect::DumpOperatorVersion::run().
std::string tflinspect::Reader::opcode_name | ( | const tflite::Operator * | op | ) | const |
Definition at line 77 of file Reader.cpp.
References mio::tflite::is_valid(), and mio::tflite::opcode_name().
Referenced by tflinspect::DumpOperators::run(), tflinspect::DumpConv2DWeight::run(), and tflinspect::DumpOperatorVersion::run().
|
inline |
Definition at line 56 of file Reader.h.
Referenced by Reader(), and tflinspect::DumpOperatorVersion::run().
|
inline |
Definition at line 59 of file Reader.h.
Referenced by tflinspect::DumpOperators::run(), tflinspect::DumpConv2DWeight::run(), and tflinspect::DumpOperatorVersion::run().
|
inline |
Definition at line 61 of file Reader.h.
Referenced by validate_onnx2circle.OnnxRunner::get_outputs(), package.infer.session::inference(), and package.infer.session::set_outputs().
bool tflinspect::Reader::select_subgraph | ( | uint32_t | subgraph | ) |
Definition at line 93 of file Reader.cpp.
References tflinspect::as_index_vector(), and flatbuffers::Vector< T >::size().
Referenced by tflinspect::DumpOperators::run(), tflinspect::DumpConv2DWeight::run(), and tflinspect::DumpOperatorVersion::run().
|
inline |