23.05
|
Go to the documentation of this file.
39 auto inputDecoder = MakeDecoder<float>(inputInfo, inputs[0]->
Map());
40 auto outputEncoder = MakeEncoder<float>(outputInfo, outputs[0]->
Map());
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers
std::vector< ITensorHandle * > m_Inputs
std::vector< unsigned int > m_Axis
Values for the dimensions to reduce.
MeanQueueDescriptor m_Data
void Execute() const override
Copyright (c) 2021 ARM Limited and Contributors.
void Reduce(const TensorInfo &inputInfo, const TensorInfo &outputInfo, Decoder< float > &input, Encoder< float > &output, const std::vector< uint32_t > axis, const ReduceOperation reduceOperation)
#define ARMNN_SCOPED_PROFILING_EVENT(backendId, name)
Contains information about TensorInfos of a layer.
LayerDescriptor m_Parameters
RefMeanWorkload(const MeanQueueDescriptor &descriptor, const WorkloadInfo &info)
std::vector< ITensorHandle * > m_Outputs
std::vector< ITensorHandle * > m_Outputs
void ExecuteAsync(ExecutionData &executionData) override
@ CpuRef
CPU Execution: Reference C++ kernels.
std::vector< ITensorHandle * > m_Inputs