#include "NeonBaseWorkload.hpp"
#include "arm_compute/runtime/Tensor.h"
#include "arm_compute/runtime/NEON/functions/NEGather.h"
#include "arm_compute/runtime/NEON/functions/NEPixelWiseMultiplication.h"
#include "arm_compute/runtime/NEON/functions/NEReductionOperation.h"
#include "arm_compute/runtime/NEON/functions/NEReshapeLayer.h"
Go to the source code of this file.
|
| | armnn |
| | Copyright (c) 2021 ARM Limited and Contributors.
|
| |
|
| arm_compute::Status | NeonGatherNdWorkloadValidate (const TensorInfo ¶msInfo, const TensorInfo &indicesInfo, const TensorInfo &outputInfo) |
| |