Home
last modified time | relevance | path

Searched refs:mTfliteInterpreter (Results 1 – 2 of 2) sorted by relevance

/test/mlts/benchmark/jni/
Drun_tflite.cpp93 tflite::InterpreterBuilder(*mTfliteModel, resolver)(&mTfliteInterpreter); in init()
94 if (!mTfliteInterpreter) { in init()
104 for (size_t node = 0; node < mTfliteInterpreter->nodes_size(); ++node) { in init()
106 mTfliteInterpreter->node_and_registration(node)->first.outputs; in init()
110 mTfliteInterpreter->SetOutputs(outputs); in init()
114 mTfliteInterpreter->SetAllowFp16PrecisionForFp32(true); in init()
121 if (mTfliteInterpreter->ModifyGraphWithDelegate( in init()
135 int input = mTfliteInterpreter->inputs()[0]; in setInput()
136 auto* input_tensor = mTfliteInterpreter->tensor(input); in setInput()
154 int output = mTfliteInterpreter->outputs()[output_index]; in saveInferenceOutput()
[all …]
Drun_tflite.h96 std::unique_ptr<tflite::Interpreter> mTfliteInterpreter; variable