34 std::unique_ptr<Decoder<float>> decoderPtr = MakeDecoder<float>(inputInfo0, inputs[0]->Map());
37 const int32_t* indicesData =
reinterpret_cast<int32_t*
>(inputs[1]->Map());
39 std::unique_ptr<Encoder<float>> encoderPtr = MakeEncoder<float>(outputInfo, outputs[0]->Map());
CPU Execution: Reference C++ kernels.
Copyright (c) 2021 ARM Limited and Contributors.
LayerDescriptor m_Parameters
#define ARMNN_SCOPED_PROFILING_EVENT(backendId, name)
std::vector< ITensorHandle * > m_Inputs
void Execute() const override
GatherQueueDescriptor m_Data
int32_t m_Axis
The axis in params to gather indices from.
std::vector< ITensorHandle * > m_Outputs
std::vector< ITensorHandle * > m_Outputs
std::vector< ITensorHandle * > m_Inputs
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers
void ExecuteAsync(WorkingMemDescriptor &workingMemDescriptor) override