Home
last modified time | relevance | path

Searched refs:m_TfLiteInterpreter (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/armnn/tests/ExecuteNetwork/
H A DTfliteExecutor.cpp17 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()
[all …]
H A DTfliteExecutor.hpp33 std::unique_ptr<Interpreter> m_TfLiteInterpreter; member in TfLiteExecutor
/aosp_15_r20/external/armnn/delegate/common/src/test/
H A DDelegateTestInterpreter.hpp38 m_TfLiteInterpreter = TfLiteInterpreterCreate(model, options); in DelegateTestInterpreter()
61 return TfLiteInterpreterAllocateTensors(m_TfLiteInterpreter); in AllocateTensors()
68 …* inputTensor = delegateTestInterpreter::GetInputTensorFromInterpreter(m_TfLiteInterpreter, index); in FillInputTensor()
77 …* inputTensor = delegateTestInterpreter::GetInputTensorFromInterpreter(m_TfLiteInterpreter, index); in FillInputTensor()
101 return TfLiteInterpreterInvoke(m_TfLiteInterpreter); in Invoke()
110 delegateTestInterpreter::GetOutputTensorFromInterpreter(m_TfLiteInterpreter, index); in GetOutputResult()
131 delegateTestInterpreter::GetOutputTensorFromInterpreter(m_TfLiteInterpreter, index); in GetOutputResult()
152 delegateTestInterpreter::GetOutputTensorFromInterpreter(m_TfLiteInterpreter, index); in GetOutputShape()
169 TfLiteInterpreter* m_TfLiteInterpreter; member in delegateTestInterpreter::DelegateTestInterpreter
/aosp_15_r20/external/armnn/delegate/opaque/src/test/
H A DDelegateTestInterpreter.cpp36 m_TfLiteInterpreter = TfLiteInterpreterCreate(tfLiteModel, options); in DelegateTestInterpreter()
62 m_TfLiteInterpreter = TfLiteInterpreterCreate(tfLiteModel, options); in DelegateTestInterpreter()
71 TfLiteInterpreterDelete(m_TfLiteInterpreter); in Cleanup()
/aosp_15_r20/external/armnn/delegate/classic/src/test/
H A DDelegateTestInterpreter.cpp34 m_TfLiteInterpreter = TfLiteInterpreterCreate(tfLiteModel, options); in DelegateTestInterpreter()
57 m_TfLiteInterpreter = TfLiteInterpreterCreate(tfLiteModel, options); in DelegateTestInterpreter()
66 TfLiteInterpreterDelete(m_TfLiteInterpreter); in Cleanup()