27 std::unique_ptr<Decoder<float>> decoderPtr = MakeDecoder<float>(inputTensorInfo,
m_Data.
m_Inputs[0]->Map());
32 int32_t* output = GetOutputTensorData<int32_t>(0,
m_Data);
#define ARMNN_SCOPED_PROFILING_EVENT(backendId, name)
LayerDescriptor m_Parameters
virtual void Execute() const override
ArgMinMaxFunction m_Function
Specify if the function is to find Min or Max.
int m_Axis
Axis to reduce across the input tensor.
const ArgMinMaxQueueDescriptor m_Data
CPU Execution: Reference C++ kernels.
std::vector< ITensorHandle * > m_Outputs
std::vector< ITensorHandle * > m_Inputs
RefArgMinMaxWorkload(const ArgMinMaxQueueDescriptor &descriptor, const WorkloadInfo &info)
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers