Home
last modified time | relevance | path

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

/test/mlts/benchmark/jni/
Drun_tflite.cpp115 tflite::InterpreterBuilder(*mTfliteModel, resolver)(&mTfliteInterpreter); in init()
116 if (!mTfliteInterpreter) { in init()
126 for (size_t node = 0; node < mTfliteInterpreter->nodes_size(); ++node) { in init()
128 mTfliteInterpreter->node_and_registration(node)->first.outputs; in init()
132 mTfliteInterpreter->SetOutputs(outputs); in init()
136 mTfliteInterpreter->SetAllowFp16PrecisionForFp32(true); in init()
142 int delegationStatus = mTfliteInterpreter->ModifyGraphWithDelegate(mTfliteNnapiDelegate.get()); in init()
160 int input = mTfliteInterpreter->inputs()[0]; in setInput()
161 auto* input_tensor = mTfliteInterpreter->tensor(input); in setInput()
179 int output = mTfliteInterpreter->outputs()[output_index]; in saveInferenceOutput()
[all …]
Drun_tflite.h131 std::unique_ptr<tflite::Interpreter> mTfliteInterpreter; variable