Searched refs:m_interpreter (Results 1 – 2 of 2) sorted by relevance
42 std::unique_ptr<tflite::Interpreter> m_interpreter; member in common::ArmnnNetworkExecutor119 tflite::InterpreterBuilder(*m_model, resolver)(&m_interpreter); in ArmnnNetworkExecutor()120 if (m_interpreter->AllocateTensors() != kTfLiteOk) in ArmnnNetworkExecutor()148 m_interpreter->ModifyGraphWithDelegate(std::move(theArmnnDelegate)); in ArmnnNetworkExecutor()156 size_t inputTensorSize = m_interpreter->input_tensor(0)->bytes; in PrepareTensors()157 auto * inputTensorPtr = m_interpreter->input_tensor(0)->data.raw; in PrepareTensors()180 if (m_interpreter->Invoke() == kTfLiteOk) in Run()187 outResults.reserve(m_interpreter->outputs().size()); in Run()188 for (int index = 0; index < m_interpreter->outputs().size(); index++) in Run()190 size_t size = m_interpreter->output_tensor(index)->bytes / sizeof(Tout); in Run()[all …]
497 tflite::InterpreterBuilder(*m_model, resolver)(&m_interpreter);501 m_interpreter->AllocateTensors();540 m_interpreter->ModifyGraphWithDelegate(std::move(theArmnnDelegate));602 if (m_interpreter->Invoke() == kTfLiteOk)