| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | GatherFixture.h | 65 …void generate_indices(U &&indices, const TensorShape &input_shape, uint32_t actual_axis, TensorSha… in generate_indices()
86 … const uint32_t actual_axis = wrap_around(axis, static_cast<int>(input_shape.num_dimensions())); in compute_target() local
125 …const uint32_t actual_axis = wrap_around(axis, static_cast<int>(input_shape.num_dimensions… in compute_reference() local
|
| D | L2NormalizeLayerFixture.h | 105 uint32_t actual_axis = wrap_around(axis, max_input_tensor_dim); in compute_reference() local
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEL2NormalizeLayer.cpp | 53 const uint32_t actual_axis = wrap_around(axis, max_input_tensor_dim); in configure() local
71 const uint32_t actual_axis = wrap_around(axis, max_input_tensor_dim); in validate() local
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLL2NormalizeLayer.cpp | 71 const uint32_t actual_axis = wrap_around(axis, max_input_tensor_dim); in configure() local
88 const uint32_t actual_axis = wrap_around(axis, max_input_tensor_dim); in validate() local
|
| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLGatherKernel.cpp | 40 const uint32_t actual_axis = wrap_around(axis, static_cast<int>(input->num_dimensions())); in validate_arguments() local
62 const uint32_t actual_axis = wrap_around(axis, static_cast<int>(input->num_dimensions())); in validate_and_configure_window() local
|
| D | CLL2NormalizeLayerKernel.cpp | 49 const uint32_t actual_axis = wrap_around(axis, max_input_tensor_dim); in validate_arguments() local
|
| /external/ComputeLibrary/src/gpu/cl/operators/ |
| D | ClSoftmax.cpp | 60 …const size_t actual_axis = static_cast<size_t>(wrap_around(info.axis, static_cast<int32_t>(src.num… in configure() local
106 …const size_t actual_axis = static_cast<size_t>(wrap_around(info.axis, static_cast<int32_t>(src.n… in validate() local
|
| /external/ComputeLibrary/src/cpu/operators/ |
| D | CpuSoftmax.cpp | 66 …const unsigned int actual_axis = static_cast<unsigned int>(wrap_around(axis, static_cast<int32_t>(… in configure() local
137 …const unsigned int actual_axis = static_cast<unsigned int>(wrap_around(axis, static_cast<int32_t>(… in validate() local
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NEL2NormalizeLayerKernel.cpp | 52 unsigned int actual_axis; member
113 const uint32_t actual_axis = wrap_around(axis, max_input_tensor_dim); in validate_arguments() local
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | Gather.cpp | 40 …<T> gather(const SimpleTensor<T> &src, const SimpleTensor<uint32_t> &indices, uint32_t actual_axis) in gather()
|
| D | SoftmaxLayer.cpp | 47 const unsigned int actual_axis = static_cast<unsigned int>(wrap_around(axis, n_dims)); in softmax_layer_generic() local
|
| /external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/operators/ |
| D | GpuSoftmax.cpp | 114 …const size_t actual_axis = static_cast<size_t>(wrap_around(attributes.axis(), static_cast<int32_… in validate_op() local
|
| /external/ComputeLibrary/arm_compute/core/utils/misc/ |
| D | ShapeCalculator.h | 1538 …ather_shape(const TensorShape &input_shape, const TensorShape &indices_shape, uint32_t actual_axis) in compute_gather_shape()
|