Searched refs:GetOutputTensorData (Results 1 – 11 of 11) sorted by relevance
/aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
H A D | RefWorkloadUtils.hpp | 43 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 D | RefArgMinMaxWorkload.cpp | 45 int32_t *output = GetOutputTensorData<int32_t>(outputs[0]); in Execute() 49 int64_t *output = GetOutputTensorData<int64_t>(outputs[0]); in Execute()
|
H A D | RefDebugWorkload.cpp | 40 T* outputData = GetOutputTensorData<T>(0, m_Data); in Execute()
|
H A D | Splitter.hpp | 71 DataType* outputData = GetOutputTensorData<DataType>(viewIdx, data); in Splitter()
|
/aosp_15_r20/external/armnn/src/dynamic/sample/ |
H A D | SampleDynamicAdditionWorkload.cpp | 28 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 D | NeonRankWorkload.hpp | 25 std::memcpy(GetOutputTensorData<void>(0, m_Data), &rank, sizeof(int32_t)); in Execute()
|
H A D | NeonWorkloadUtils.hpp | 178 DataType* GetOutputTensorData(unsigned int idx, const PayloadType& data) in GetOutputTensorData() function
|
/aosp_15_r20/external/armnn/src/backends/cl/workloads/ |
H A D | ClRankWorkload.hpp | 25 std::memcpy(GetOutputTensorData<void>(0, m_Data), &rank, sizeof(int32_t)); in Execute()
|
H A D | ClWorkloadUtils.hpp | 173 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 D | HelloTFLite.cs | 78 interpreter.GetOutputTensorData(0, outputs); in Update()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/examples/unity/TensorFlowLitePlugin/Assets/TensorFlowLite/SDK/Scripts/ |
H A D | Interpreter.cs | 118 public void GetOutputTensorData(int outputTensorIndex, Array outputTensorData) { in GetOutputTensorData() method in TensorFlowLite.Interpreter
|