Lines Matching refs:mTfliteInterpreter

93   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()
155 auto* output_tensor = mTfliteInterpreter->tensor(output); in saveInferenceOutput()
163 int output = mTfliteInterpreter->outputs()[output_index]; in getOutputError()
164 auto* output_tensor = mTfliteInterpreter->tensor(output); in getOutputError()
175 uint8_t* output_raw = mTfliteInterpreter->typed_tensor<uint8_t>(output); in getOutputError()
186 float* output_raw = mTfliteInterpreter->typed_tensor<float>(output); in getOutputError()
204 int input = mTfliteInterpreter->inputs()[0]; in resizeInputTensors()
205 mTfliteInterpreter->ResizeInputTensor(input, shape); in resizeInputTensors()
206 if (mTfliteInterpreter->AllocateTensors() != kTfLiteOk) { in resizeInputTensors()
215 auto status = mTfliteInterpreter->Invoke(); in runInference()
225 auto status = mTfliteInterpreter->ResetVariableTensors(); in resetStates()
259 int input = mTfliteInterpreter->inputs()[0]; in benchmark()
260 auto* input_tensor = mTfliteInterpreter->tensor(input); in benchmark()
281 size_t outputsCount = mTfliteInterpreter->outputs().size(); in benchmark()
338 for (int tensor = 0; tensor < mTfliteInterpreter->tensors_size(); in dumpAllLayers()
340 auto* output_tensor = mTfliteInterpreter->tensor(tensor); in dumpAllLayers()