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);
CPU Execution: Reference C++ kernels.
const ArgMinMaxQueueDescriptor m_Data
ArgMinMaxFunction m_Function
Specify if the function is to find Min or Max.
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers
Copyright (c) 2020 ARM Limited.
LayerDescriptor m_Parameters
#define ARMNN_SCOPED_PROFILING_EVENT(backendId, name)
virtual void Execute() const override
std::vector< ITensorHandle * > m_Outputs
RefArgMinMaxWorkload(const ArgMinMaxQueueDescriptor &descriptor, const WorkloadInfo &info)
Contains information about inputs and outputs to a layer.
int m_Axis
Axis to reduce across the input tensor.
std::vector< ITensorHandle * > m_Inputs