23.02
|
Go to the documentation of this file.
40 std::unique_ptr<Decoder<float>> decoderPtr = MakeDecoder<float>(inputInfo, inputs[0]->
Map());
43 std::unique_ptr<Encoder<float>> encoderPtr = MakeEncoder<float>(outputInfo, outputs[0]->
Map());
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers
ReduceOperation m_ReduceOperation
Specifies the reduction operation to execute.
std::vector< ITensorHandle * > m_Inputs
std::vector< uint32_t > m_vAxis
The indices of the dimensions to reduce.
ReduceQueueDescriptor m_Data
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.
RefReduceWorkload(const ReduceQueueDescriptor &descriptor, const WorkloadInfo &info)
LayerDescriptor m_Parameters
std::vector< ITensorHandle * > m_Outputs
std::vector< ITensorHandle * > m_Outputs
void Execute() const override
void ExecuteAsync(ExecutionData &executionData) override
@ CpuRef
CPU Execution: Reference C++ kernels.
std::vector< ITensorHandle * > m_Inputs