Lines Matching refs:m_TfLiteInterpreter
17 m_TfLiteInterpreter = std::make_unique<Interpreter>(); in TfLiteExecutor()
21 if (builder(&m_TfLiteInterpreter) != kTfLiteOk) in TfLiteExecutor()
25 if (m_TfLiteInterpreter->AllocateTensors() != kTfLiteOk) in TfLiteExecutor()
39 if (m_TfLiteInterpreter->ModifyGraphWithDelegate(std::move(theArmnnDelegate)) != kTfLiteOk) in TfLiteExecutor()
49 const size_t numInputs = m_TfLiteInterpreter->inputs().size(); in TfLiteExecutor()
57 int input = m_TfLiteInterpreter->inputs()[inputIndex]; in TfLiteExecutor()
58 const auto& inputName = m_TfLiteInterpreter->tensor(input)->name; in TfLiteExecutor()
61 if (!tflite::IsConstantTensor(m_TfLiteInterpreter->tensor(input))) in TfLiteExecutor()
63 TfLiteIntArray* inputDims = m_TfLiteInterpreter->tensor(input)->dims; in TfLiteExecutor()
71 const auto& dataType = m_TfLiteInterpreter->tensor(input)->type; in TfLiteExecutor()
77 auto inputData = m_TfLiteInterpreter->typed_tensor<float>(input); in TfLiteExecutor()
83 auto inputData = m_TfLiteInterpreter->typed_tensor<int32_t>(input); in TfLiteExecutor()
89 auto inputData = m_TfLiteInterpreter->typed_tensor<uint8_t>(input); in TfLiteExecutor()
95 auto inputData = m_TfLiteInterpreter->typed_tensor<int16_t>(input); in TfLiteExecutor()
101 auto inputData = m_TfLiteInterpreter->typed_tensor<int8_t>(input); in TfLiteExecutor()
127 status = m_TfLiteInterpreter->Invoke(); in Execute()
133 …for (unsigned int outputIndex = 0; outputIndex < m_TfLiteInterpreter->outputs().size(); ++outputIn… in Execute()
135 auto tfLiteDelegateOutputId = m_TfLiteInterpreter->outputs()[outputIndex]; in Execute()
136 … TfLiteIntArray* outputDims = m_TfLiteInterpreter->tensor(tfLiteDelegateOutputId)->dims; in Execute()
159 std::cout << m_TfLiteInterpreter->tensor(tfLiteDelegateOutputId)->name << ": "; in Execute()
160 results.push_back(m_TfLiteInterpreter->tensor(tfLiteDelegateOutputId)->allocation); in Execute()
162 switch (m_TfLiteInterpreter->tensor(tfLiteDelegateOutputId)->type) in Execute()
167 auto tfLiteDelegateOutputData = m_TfLiteInterpreter->typed_tensor<float>( in Execute()
178 auto tfLiteDelegateOutputData = m_TfLiteInterpreter->typed_tensor<int32_t>( in Execute()
188 auto tfLiteDelegateOutputData = m_TfLiteInterpreter->typed_tensor<uint8_t>( in Execute()
198 auto tfLiteDelegateOutputData = m_TfLiteInterpreter->typed_tensor<int8_t>( in Execute()
208 auto tfLiteDelegateOutputData = m_TfLiteInterpreter->typed_tensor<bool>( in Execute()
233 …for (unsigned int outputIndex = 0; outputIndex < m_TfLiteInterpreter->outputs().size(); ++outputIn… in CompareAndPrintResult()
235 auto tfLiteDelegateOutputId = m_TfLiteInterpreter->outputs()[outputIndex]; in CompareAndPrintResult()
236 size_t size = m_TfLiteInterpreter->tensor(tfLiteDelegateOutputId)->bytes; in CompareAndPrintResult()
237 …double result = ComputeByteLevelRMSE(m_TfLiteInterpreter->tensor(tfLiteDelegateOutputId)->allocati… in CompareAndPrintResult()