Home
last modified time | relevance | path

Searched refs:GetOutputTensorData (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DRefWorkloadUtils.hpp43 DataType* GetOutputTensorData(unsigned int idx, const PayloadType& data) in GetOutputTensorData() function
50 DataType* GetOutputTensorData(ITensorHandle* tensorHandle) in GetOutputTensorData() function
64 return GetOutputTensorData<float>(idx, data); in GetOutputTensorDataFloat()
76 return GetOutputTensorData<Half>(idx, data); in GetOutputTensorDataHalf()
88 return GetOutputTensorData<BFloat16>(idx, data); in GetOutputTensorDataBFloat16()
H A DRefArgMinMaxWorkload.cpp45 int32_t *output = GetOutputTensorData<int32_t>(outputs[0]); in Execute()
49 int64_t *output = GetOutputTensorData<int64_t>(outputs[0]); in Execute()
H A DRefDebugWorkload.cpp40 T* outputData = GetOutputTensorData<T>(0, m_Data); in Execute()
H A DSplitter.hpp71 DataType* outputData = GetOutputTensorData<DataType>(viewIdx, data); in Splitter()
/aosp_15_r20/external/armnn/src/dynamic/sample/
H A DSampleDynamicAdditionWorkload.cpp28 float* GetOutputTensorData(unsigned int idx, const armnn::AdditionQueueDescriptor& data) in GetOutputTensorData() function
46 float* outputData = GetOutputTensorData(0, m_Data); in Execute()
/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonRankWorkload.hpp25 std::memcpy(GetOutputTensorData<void>(0, m_Data), &rank, sizeof(int32_t)); in Execute()
H A DNeonWorkloadUtils.hpp178 DataType* GetOutputTensorData(unsigned int idx, const PayloadType& data) in GetOutputTensorData() function
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClRankWorkload.hpp25 std::memcpy(GetOutputTensorData<void>(0, m_Data), &rank, sizeof(int32_t)); in Execute()
H A DClWorkloadUtils.hpp173 DataType* GetOutputTensorData(unsigned int idx, const PayloadType& data) in GetOutputTensorData() function
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/examples/unity/TensorFlowLitePlugin/Assets/TensorFlowLite/Examples/HelloTFLite/Scripts/
H A DHelloTFLite.cs78 interpreter.GetOutputTensorData(0, outputs); in Update()
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/examples/unity/TensorFlowLitePlugin/Assets/TensorFlowLite/SDK/Scripts/
H A DInterpreter.cs118 public void GetOutputTensorData(int outputTensorIndex, Array outputTensorData) { in GetOutputTensorData() method in TensorFlowLite.Interpreter