18 const arm_compute::TensorInfo aclInput = BuildArmComputeTensorInfo(input);
19 const arm_compute::TensorInfo aclIndices = BuildArmComputeTensorInfo(indices);
20 const arm_compute::TensorInfo aclOutput = BuildArmComputeTensorInfo(output);
24 return arm_compute::NEGather::validate(&aclInput, &aclIndices, &aclOutput, aclAxis);
33 arm_compute::ITensor& input = PolymorphicDowncast<IAclTensorHandle*>(
m_Data.
m_Inputs[0])->GetTensor();
34 arm_compute::ITensor& indices = PolymorphicDowncast<IAclTensorHandle*>(
m_Data.
m_Inputs[1])->GetTensor();
35 arm_compute::ITensor& output = PolymorphicDowncast<IAclTensorHandle*>(
m_Data.
m_Outputs[0])->GetTensor();
39 m_Layer.configure(&input, &indices, &output, aclAxis);
arm_compute::Status NeonGatherWorkloadValidate(const TensorInfo &input, const TensorInfo &indices, const TensorInfo &output, const GatherDescriptor &descriptor)
int ComputeAclAxis(const int &armnnAxis, const armnn::TensorInfo &tensor)
Function to convert ArmNN axis (left to right) to ACL axis (right to left) ranging from [-rank...
const GatherQueueDescriptor m_Data
#define ARMNN_SCOPED_PROFILING_EVENT_NEON(name)
void ValidateInputsOutputs(const std::string &descName, unsigned int numExpectedIn, unsigned int numExpectedOut) const
Copyright (c) 2020 ARM Limited.
LayerDescriptor m_Parameters
std::vector< TensorInfo > m_InputTensorInfos
A GatherDescriptor for the GatherLayer.
NeonGatherWorkload(const GatherQueueDescriptor &descriptor, const WorkloadInfo &info)
int32_t m_Axis
The axis in params to gather indices from.
std::vector< ITensorHandle * > m_Outputs
Contains information about inputs and outputs to a layer.
virtual void Execute() const override
std::vector< ITensorHandle * > m_Inputs