Home
last modified time | relevance | path

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

/test/mlts/benchmark/jni/
Drun_tflite.cpp117 tflite::InterpreterBuilder(*mTfliteModel, resolver)(&mTfliteInterpreter); in init()
118 if (!mTfliteInterpreter) { in init()
128 for (size_t node = 0; node < mTfliteInterpreter->nodes_size(); ++node) { in init()
130 mTfliteInterpreter->node_and_registration(node)->first.outputs; in init()
134 mTfliteInterpreter->SetOutputs(outputs); in init()
138 mTfliteInterpreter->SetAllowFp16PrecisionForFp32(true); in init()
146 … int delegationStatus = mTfliteInterpreter->ModifyGraphWithDelegate(mTfliteNnapiDelegate.get()); in init()
160 if (mTfliteInterpreter->ModifyGraphWithDelegate(mGpuDelegate) != in init()
192 int input = mTfliteInterpreter->inputs()[0]; in setInput()
193 auto* input_tensor = mTfliteInterpreter->tensor(input); in setInput()
[all …]
Drun_tflite.h143 std::unique_ptr<tflite::Interpreter> mTfliteInterpreter; variable