10 #include <arm_compute/runtime/CL/functions/CLGather.h> 15 const TensorInfo& indices,
16 const TensorInfo& output,
17 const GatherDescriptor& descriptor);
26 mutable arm_compute::CLGather m_Layer;
ClGatherWorkload(const GatherQueueDescriptor &descriptor, const WorkloadInfo &info)
Copyright (c) 2020 ARM Limited.
arm_compute::Status ClGatherWorkloadValidate(const TensorInfo &input, const TensorInfo &indices, const TensorInfo &output, const GatherDescriptor &descriptor)
Contains information about inputs and outputs to a layer.
void Execute() const override