24.04
|
#include "Utils.h"
#include "arm_compute/function_info/ScatterInfo.h"
#include "tests/SimpleTensor.h"
Go to the source code of this file.
Namespaces | |
arm_compute | |
Copyright (c) 2017-2024 Arm Limited. | |
arm_compute::test | |
arm_compute::test::validation | |
arm_compute::test::validation::reference | |
Functions | |
template<typename T > | |
SimpleTensor< T > | scatter_layer_internal (const SimpleTensor< T > &src, const SimpleTensor< T > &updates, const SimpleTensor< uint32_t > &indices, const TensorShape &out_shape, const ScatterInfo &info) |
template<typename T > | |
SimpleTensor< T > | scatter_layer (const SimpleTensor< T > &src, const SimpleTensor< T > &updates, const SimpleTensor< uint32_t > &indices, const TensorShape &out_shape, const ScatterInfo &info) |