| /external/android-nn-driver/test/ |
| D | Concat.cpp | 152 int32_t axis = 0; in BOOST_DATA_TEST_CASE() local
164 int32_t axis = 0; in BOOST_DATA_TEST_CASE() local
184 int32_t axis = 1; in BOOST_DATA_TEST_CASE() local
196 int32_t axis = 1; in BOOST_DATA_TEST_CASE() local
216 int32_t axis = 1; in BOOST_DATA_TEST_CASE() local
232 int32_t axis = 2; in BOOST_DATA_TEST_CASE() local
244 int32_t axis = 2; in BOOST_DATA_TEST_CASE() local
264 int32_t axis = 2; in BOOST_DATA_TEST_CASE() local
280 int32_t axis = 3; in BOOST_DATA_TEST_CASE() local
292 int32_t axis = 3; in BOOST_DATA_TEST_CASE() local
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sort_ops.py | 35 def sort(values, axis=-1, direction='ASCENDING', name=None): argument
92 def argsort(values, axis=-1, direction='ASCENDING', stable=False, name=None): argument
157 def _sort_or_argsort(values, axis, direction, return_argsort): argument
188 def _descending_sort(values, axis, return_argsort=False): argument
243 def _ascending_sort(values, axis, return_argsort=False): argument
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_shape.py | 195 def dimension_size(self, axis): argument
205 def is_ragged(self, axis): argument
294 def broadcast_dimension(self, axis, lengths): argument
374 def num_slices_in_dimension(self, axis): argument
383 def _broadcast_uniform_partitioned_dimension(self, axis, lengths): argument
412 def _broadcast_inner_dimension_to_uniform(self, axis, length): argument
425 def _broadcast_inner_dimension_to_ragged(self, axis, lengths): argument
603 def _ragged_tile_axis(rt_input, axis, repeats, row_splits_dtype): argument
|
| D | ragged_math_ops.py | 454 axis, argument
584 def reduce_sum(input_tensor, axis=None, keepdims=None, name=None): argument
596 def reduce_prod(input_tensor, axis=None, keepdims=None, name=None): argument
607 def reduce_min(input_tensor, axis=None, keepdims=None, name=None): argument
618 def reduce_max(input_tensor, axis=None, keepdims=None, name=None): argument
629 def reduce_mean(input_tensor, axis=None, keepdims=None, name=None): argument
650 def reduce_variance(input_tensor, axis=None, keepdims=False, name=None): argument
660 def reduce_std(input_tensor, axis=None, keepdims=False, name=None): argument
672 def reduce_all(input_tensor, axis=None, keepdims=None, name=None): argument
680 def reduce_any(input_tensor, axis=None, keepdims=None, name=None): argument
[all …]
|
| D | ragged_gather_ops.py | 39 axis=None, argument
114 def _gather(params, indices, axis, batch_dims): argument
176 def _batch_gather(params, indices, axis, batch_dims): argument
258 def _axis_gather(params, indices, axis): argument
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_array_ops.py | 376 def all(a, axis=None, keepdims=None): # pylint: disable=redefined-builtin argument
382 def any(a, axis=None, keepdims=None): # pylint: disable=redefined-builtin argument
388 def compress(condition, a, axis=None): # pylint: disable=redefined-outer-name,missing-function-doc… argument
435 def cumprod(a, axis=None, dtype=None): # pylint: disable=missing-docstring argument
451 def cumsum(a, axis=None, dtype=None): # pylint: disable=missing-docstring argument
480 axis=None, argument
550 def size(x, axis=None): # pylint: disable=missing-docstring argument
564 def sum(a, axis=None, dtype=None, keepdims=None): # pylint: disable=redefined-builtin argument
575 def prod(a, axis=None, dtype=None, keepdims=None): argument
586 def mean(a, axis=None, dtype=None, out=None, keepdims=None): argument
[all …]
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | SoftmaxLayerFixture.h | 50 …TensorShape shape, DataType data_type, QuantizationInfo quantization_info, float beta, size_t axis) in setup()
75 QuantizationInfo quantization_info, float beta, int32_t axis) in compute_target()
105 QuantizationInfo quantization_info, float beta, int32_t axis) in compute_reference()
126 void setup(TensorShape shape, DataType data_type, float beta, size_t axis) in setup()
141 …TensorShape shape, DataType data_type, QuantizationInfo quantization_info, float beta, size_t axis) in setup()
|
| D | ArgMinMaxFixture.h | 50 …void setup(TensorShape shape, DataType data_type, int axis, ReductionOperation op, QuantizationInf… in setup()
96 …TensorType compute_target(TensorShape &src_shape, DataType data_type, int axis, ReductionOperation… in compute_target()
125 …SimpleTensor<int32_t> compute_reference(TensorShape &src_shape, DataType data_type, int axis, Redu… in compute_reference()
146 …void setup(const TensorShape &shape, DataType data_type, int axis, ReductionOperation op, Quantiza… in setup()
157 void setup(const TensorShape &shape, DataType data_type, int axis, ReductionOperation op) in setup()
|
| D | SplitFixture.h | 52 void setup(TensorShape shape, unsigned int axis, unsigned int splits, DataType data_type) in setup()
65 …std::vector<TensorType> compute_target(const TensorShape &shape, unsigned int axis, unsigned int s… in compute_target()
110 …std::vector<SimpleTensor<T>> compute_reference(const TensorShape &shape, unsigned int axis, unsign… in compute_reference()
154 …void setup(TensorShape shape, unsigned int axis, std::vector<TensorShape> split_shapes, DataType d… in setup()
167 …std::vector<TensorType> compute_target(TensorShape shape, unsigned int axis, std::vector<TensorSha… in compute_target()
219 …std::vector<SimpleTensor<T>> compute_reference(TensorShape shape, unsigned int axis, std::vector<T… in compute_reference()
|
| D | ReduceMeanFixture.h | 51 …void setup(TensorShape shape, DataType data_type, Coordinates axis, bool keep_dims, QuantizationIn… in setup()
75 …TensorType compute_target(TensorShape &src_shape, DataType data_type, Coordinates axis, bool keep_… in compute_target()
105 …SimpleTensor<T> compute_reference(TensorShape &src_shape, DataType data_type, Coordinates axis, bo… in compute_reference()
144 …void setup(TensorShape shape, DataType data_type, Coordinates axis, bool keep_dims, QuantizationIn… in setup()
155 void setup(TensorShape shape, DataType data_type, Coordinates axis, bool keep_dims) in setup()
|
| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLGatherKernel.cpp | 37 …rguments(const ITensorInfo *input, const ITensorInfo *indices, const ITensorInfo *output, int axis) in validate_arguments()
59 …idate_and_configure_window(ITensorInfo *input, ITensorInfo *indices, ITensorInfo *output, int axis) in validate_and_configure_window()
81 …herKernel::configure(const ICLTensor *input, const ICLTensor *indices, ICLTensor *output, int axis) in configure()
86 …xt &compile_context, const ICLTensor *input, const ICLTensor *indices, ICLTensor *output, int axis) in configure()
112 …validate(const ITensorInfo *input, const ITensorInfo *indices, const ITensorInfo *output, int axis) in validate()
|
| D | CLStackLayerKernel.cpp | 45 Status validate_arguments(const ITensorInfo *input, unsigned int axis, unsigned int idx_input, unsi… in validate_arguments()
64 std::pair<Status, Window> validate_and_configure_window(ITensorInfo *input, unsigned int axis, unsi… in validate_and_configure_window()
81 void CLStackLayerKernel::configure(const ICLTensor *input, unsigned int axis, unsigned int idx_inpu… in configure()
86 …re(const CLCompileContext &compile_context, const ICLTensor *input, unsigned int axis, unsigned in… in configure()
114 Status CLStackLayerKernel::validate(const ITensorInfo *input, unsigned int axis, unsigned int idx_i… in validate()
|
| D | CLReverseKernel.cpp | 40 …us validate_arguments(const ITensorInfo *input, const ITensorInfo *output, const ITensorInfo *axis) in validate_arguments()
66 void CLReverseKernel::configure(const ICLTensor *input, ICLTensor *output, const ICLTensor *axis) in configure()
71 …LCompileContext &compile_context, const ICLTensor *input, ICLTensor *output, const ICLTensor *axis) in configure()
114 …verseKernel::validate(const ITensorInfo *input, const ITensorInfo *output, const ITensorInfo *axis) in validate()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | scan_ops_test.py | 33 def numpy_reverse(x, axis): argument
44 def handle_options(func, x, axis, exclusive, reverse): argument
80 def _compare(self, x, axis, exclusive, reverse): argument
87 def _compareAll(self, x, axis): argument
169 def _compareGradient(self, shape, axis, exclusive, reverse): argument
213 def _compare(self, x, axis, exclusive, reverse): argument
220 def _compareAll(self, x, axis): argument
295 def _compareGradient(self, shape, axis, exclusive, reverse): argument
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | scan_ops_test.py | 33 def numpy_reverse(x, axis): argument
44 def handle_options(func, x, axis, exclusive, reverse): argument
80 def _compare(self, x, axis, exclusive, reverse): argument
89 def _compareAll(self, x, axis): argument
159 def _compare(self, x, axis, exclusive, reverse): argument
168 def _compareAll(self, x, axis): argument
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NEStackLayerKernel.cpp | 44 Status validate_arguments(const ITensorInfo *input, unsigned int axis, unsigned int idx_input, unsi… in validate_arguments()
63 std::pair<Status, Window> validate_and_configure_window(ITensorInfo *input, unsigned int axis, unsi… in validate_and_configure_window()
74 inline Coordinates shift_from_axis_and_replace_coordinate(const Coordinates &id, unsigned int axis,… in shift_from_axis_and_replace_coordinate()
92 void NEStackLayerKernel::configure(const ITensor *input, unsigned int axis, unsigned int idx_input,… in configure()
109 Status NEStackLayerKernel::validate(const ITensorInfo *input, unsigned int axis, unsigned int idx_i… in validate()
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLReverse.cpp | 32 void CLReverse::configure(const ICLTensor *input, ICLTensor *output, const ICLTensor *axis) in configure()
37 …LCompileContext &compile_context, const ICLTensor *input, ICLTensor *output, const ICLTensor *axis) in configure()
44 …s CLReverse::validate(const ITensorInfo *input, const ITensorInfo *output, const ITensorInfo *axis) in validate()
|
| D | CLConcatenateLayer.cpp | 53 …compile_context, const std::vector<ITensorInfo *> &inputs_vector, ITensorInfo *output, size_t axis) in configure()
147 …date(const std::vector<const ITensorInfo *> &inputs_vector, const ITensorInfo *output, size_t axis) in validate()
261 unsigned int axis{ 0 }; member
276 …nateLayer::configure(std::vector<const ICLTensor *> &inputs_vector, ICLTensor *output, size_t axis) in configure()
281 …xt &compile_context, std::vector<const ICLTensor *> &inputs_vector, ICLTensor *output, size_t axis) in configure()
300 …date(const std::vector<const ITensorInfo *> &inputs_vector, const ITensorInfo *output, size_t axis) in validate()
|
| D | CLGather.cpp | 32 … CLGather::configure(const ICLTensor *input, const ICLTensor *indices, ICLTensor *output, int axis) in configure()
37 …xt &compile_context, const ICLTensor *input, const ICLTensor *indices, ICLTensor *output, int axis) in configure()
44 …validate(const ITensorInfo *input, const ITensorInfo *indices, const ITensorInfo *output, int axis) in validate()
|
| D | CLUnstack.cpp | 36 inline unsigned int wrap_axis(int axis, const ITensorInfo *const tensor) in wrap_axis()
62 …Unstack::configure(const ICLTensor *input, const std::vector<ICLTensor *> &output_vector, int axis) in configure()
67 … &compile_context, const ICLTensor *input, const std::vector<ICLTensor *> &output_vector, int axis) in configure()
95 …tack::validate(const ITensorInfo *input, const std::vector<ITensorInfo *> &output_vector, int axis) in validate()
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEConcatenateLayer.cpp | 50 …:configure(const std::vector<const ITensorInfo *> &inputs_vector, ITensorInfo *output, size_t axis) in configure()
104 …date(const std::vector<const ITensorInfo *> &inputs_vector, const ITensorInfo *output, size_t axis) in validate()
179 unsigned int axis{ 0 }; member
194 …ncatenateLayer::configure(std::vector<const ITensor *> inputs_vector, ITensor *output, size_t axis) in configure()
213 …date(const std::vector<const ITensorInfo *> &inputs_vector, const ITensorInfo *output, size_t axis) in validate()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | shape.cc | 42 Axis axis; member
54 std::string ToString(Axis axis) { in ToString()
118 int GetAxisIndex(Layout layout, Axis axis) { in GetAxisIndex()
122 bool HasAxis(Layout layout, Axis axis) { in HasAxis()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | expand_dims.cc | 34 int axis, TfLiteTensor* output) { in ExpandTensorDim()
57 const TfLiteTensor& axis, int* axis_value) { in GetAxisValueFromTensor()
78 const TfLiteTensor* axis; in Prepare() local
99 const TfLiteTensor* axis; in Eval() local
|
| /external/ComputeLibrary/src/graph/nodes/ |
| D | L2NormalizeLayerNode.cpp | 38 L2NormalizeLayerNode::L2NormalizeLayerNode(int axis) in L2NormalizeLayerNode()
43 L2NormalizeLayerNode::L2NormalizeLayerNode(int axis, float epsilon) in L2NormalizeLayerNode()
75 int L2NormalizeLayerNode::axis() const in axis() function in arm_compute::graph::L2NormalizeLayerNode
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | reduce_test_util.cc | 32 const std::set<tflite::gpu::Axis> axis{Axis::HEIGHT, Axis::WIDTH}; in MeanHWTest() local
59 const std::set<tflite::gpu::Axis> axis{Axis::CHANNELS}; in ReduceSumChannelsTest() local
86 const std::set<tflite::gpu::Axis> axis{Axis::CHANNELS}; in ReduceProductChannelsTest() local
114 const std::set<tflite::gpu::Axis> axis{Axis::CHANNELS}; in ReduceMaxChannelsTest() local
142 const std::set<tflite::gpu::Axis> axis{Axis::CHANNELS}; in ReduceMinChannelsTest() local
|