38 std::unique_ptr<Decoder<float>> decoderPtr = MakeDecoder<float>(inputTensorInfo, inputs[0]->Map());
44 int32_t *output = GetOutputTensorData<int32_t>(outputs[0]);
48 int64_t *output = GetOutputTensorData<int64_t>(outputs[0]);
CPU Execution: Reference C++ kernels.
ArgMinMaxFunction m_Function
Specify if the function is to find Min or Max.
Copyright (c) 2021 ARM Limited and Contributors.
LayerDescriptor m_Parameters
#define ARMNN_SCOPED_PROFILING_EVENT(backendId, name)
void ExecuteAsync(WorkingMemDescriptor &workingMemDescriptor) override
std::vector< ITensorHandle * > m_Inputs
void Execute() const override
ArgMinMaxQueueDescriptor m_Data
DataType GetDataType() const
std::vector< ITensorHandle * > m_Outputs
RefArgMinMaxWorkload(const ArgMinMaxQueueDescriptor &descriptor, const WorkloadInfo &info)
std::vector< ITensorHandle * > m_Outputs
Contains information about inputs and outputs to a layer.
int m_Axis
Axis to reduce across the input tensor.
std::vector< ITensorHandle * > m_Inputs
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers