ArmNN
 22.08
ClGatherNdWorkload.hpp File Reference
#include "ClBaseWorkload.hpp"
#include "arm_compute/runtime/Tensor.h"
#include "arm_compute/runtime/CL/functions/CLGather.h"
#include "arm_compute/runtime/CL/functions/CLPixelWiseMultiplication.h"
#include "arm_compute/runtime/CL/functions/CLReductionOperation.h"
#include "arm_compute/runtime/CL/functions/CLReshapeLayer.h"

Go to the source code of this file.

Classes

class  ClGatherNdWorkload
 

Namespaces

 armnn
 Copyright (c) 2021 ARM Limited and Contributors.
 

Functions

arm_compute::Status ClGatherNdWorkloadValidate (const TensorInfo &paramsInfo, const TensorInfo &indicesInfo, const TensorInfo &outputInfo)