| /external/android-nn-driver/test/ |
| D | Concat.cpp | 138 int32_t axis = 0; in SimpleConcatAxis0() local
149 int32_t axis = 0; in ConcatAxis0NoInterleave() local
169 int32_t axis = 1; in SimpleConcatAxis1() local
181 int32_t axis = 1; in ConcatAxis1NoInterleave() local
201 int32_t axis = 1; in SimpleConcatAxis1DoInterleave() local
217 int32_t axis = 2; in SimpleConcatAxis2() local
229 int32_t axis = 2; in ConcatAxis2NoInterleave() local
249 int32_t axis = 2; in SimpleConcatAxis2DoInterleave() local
265 int32_t axis = 3; in SimpleConcatAxis3() local
277 int32_t axis = 3; in SimpleConcatAxis3DoInterleave() local
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sort_ops.py | 31 def sort(values, axis=-1, direction='ASCENDING', name=None): argument
88 def argsort(values, axis=-1, direction='ASCENDING', stable=False, name=None): argument
153 def _sort_or_argsort(values, axis, direction, return_argsort): argument
186 def _descending_sort(values, axis, return_argsort=False): argument
241 def _ascending_sort(values, axis, return_argsort=False): argument
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_shape.py | 191 def dimension_size(self, axis): argument
201 def is_ragged(self, axis): argument
291 def broadcast_dimension(self, axis, lengths): argument
371 def num_slices_in_dimension(self, axis): argument
380 def _broadcast_uniform_partitioned_dimension(self, axis, lengths): argument
410 def _broadcast_inner_dimension_to_uniform(self, axis, length): argument
424 def _broadcast_inner_dimension_to_ragged(self, axis, lengths): argument
602 def _ragged_tile_axis(rt_input, axis, repeats, row_splits_dtype): argument
|
| D | ragged_gather_ops.py | 37 axis=None, argument
112 def _gather(params, indices, axis, batch_dims): argument
174 def _batch_gather(params, indices, axis, batch_dims): argument
256 def _axis_gather(params, indices, axis): argument
330 axis=0, argument
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | SoftmaxLayerFixture.h | 50 …TensorShape shape, DataType data_type, QuantizationInfo quantization_info, float beta, size_t axis) in setup()
84 QuantizationInfo quantization_info, float beta, int32_t axis) in compute_target()
114 QuantizationInfo quantization_info, float beta, int32_t axis) in compute_reference()
135 void setup(TensorShape shape, DataType data_type, float beta, size_t axis) in setup()
150 …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()
101 …TensorType compute_target(TensorShape &src_shape, DataType data_type, int axis, ReductionOperation… in compute_target()
130 …SimpleTensor<int32_t> compute_reference(TensorShape &src_shape, DataType data_type, int axis, Redu… in compute_reference()
151 …void setup(const TensorShape &shape, DataType data_type, int axis, ReductionOperation op, Quantiza… in setup()
162 void setup(const TensorShape &shape, DataType data_type, int axis, ReductionOperation op) in setup()
|
| D | SplitFixture.h | 51 void setup(TensorShape shape, unsigned int axis, unsigned int splits, DataType data_type) in setup()
64 …std::vector<TensorType> compute_target(const TensorShape &shape, unsigned int axis, unsigned int s… in compute_target()
109 …std::vector<SimpleTensor<T>> compute_reference(const TensorShape &shape, unsigned int axis, unsign… in compute_reference()
153 …void setup(TensorShape shape, unsigned int axis, std::vector<TensorShape> split_shapes, DataType d… in setup()
166 …std::vector<TensorType> compute_target(TensorShape shape, unsigned int axis, std::vector<TensorSha… in compute_target()
218 …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()
84 …TensorType compute_target(TensorShape &src_shape, DataType data_type, Coordinates axis, bool keep_… in compute_target()
114 …SimpleTensor<T> compute_reference(TensorShape &src_shape, DataType data_type, Coordinates axis, bo… in compute_reference()
159 …void setup(TensorShape shape, DataType data_type, Coordinates axis, bool keep_dims, QuantizationIn… in setup()
170 void setup(TensorShape shape, DataType data_type, Coordinates axis, bool keep_dims) in setup()
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_array_ops.py | 367 def all(a, axis=None, keepdims=None): # pylint: disable=redefined-builtin argument
373 def any(a, axis=None, keepdims=None): # pylint: disable=redefined-builtin argument
379 def compress(condition, a, axis=None): # pylint: disable=redefined-outer-name,missing-function-doc… argument
426 def cumprod(a, axis=None, dtype=None): # pylint: disable=missing-docstring argument
442 def cumsum(a, axis=None, dtype=None): # pylint: disable=missing-docstring argument
471 axis=None, argument
541 def size(x, axis=None): # pylint: disable=missing-docstring argument
555 def sum(a, axis=None, dtype=None, keepdims=None): # pylint: disable=redefined-builtin argument
566 def prod(a, axis=None, dtype=None, keepdims=None): argument
577 def mean(a, axis=None, dtype=None, out=None, keepdims=None): argument
[all …]
|
| /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()
114 …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()
82 void CLStackLayerKernel::configure(const ICLTensor *input, unsigned int axis, unsigned int idx_inpu… in configure()
87 …re(const CLCompileContext &compile_context, const ICLTensor *input, unsigned int axis, unsigned in… in configure()
115 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()
67 void CLReverseKernel::configure(const ICLTensor *input, ICLTensor *output, const ICLTensor *axis) in configure()
72 …LCompileContext &compile_context, const ICLTensor *input, ICLTensor *output, const ICLTensor *axis) in configure()
117 …verseKernel::validate(const ITensorInfo *input, const ITensorInfo *output, const ITensorInfo *axis) in validate()
|
| /external/ComputeLibrary/src/dynamic_fusion/sketch/attributes/ |
| D | SoftmaxAttributes.cpp | 55 SoftmaxAttributes &SoftmaxAttributes::axis(int axis) in axis() function in arm_compute::experimental::dynamic_fusion::SoftmaxAttributes
61 int SoftmaxAttributes::axis() const in axis() function in arm_compute::experimental::dynamic_fusion::SoftmaxAttributes
|
| /external/crosvm/devices/tests/data/input/ |
| D | example_custom_multitouchscreen_config.json | 27 "axis": "ABS_MT_SLOT", string
34 "axis": "ABS_MT_TRACKING_ID", string
41 "axis": "ABS_MT_POSITION_X", string
48 "axis": "ABS_MT_POSITION_Y", string
55 "axis": "ABS_X", string
62 "axis": "ABS_Y", string
|
| /external/pytorch/torch/_numpy/ |
| D | _util.py | 56 def normalize_axis_tuple(axis, ndim, argname=None, allow_duplicate=False): argument
101 def allow_only_single_axis(axis): argument
109 def expand_shape(arr_shape, axis): argument
120 def apply_keepdims(tensor, axis, ndim): argument
131 def axis_none_flatten(*tensors, axis=None): argument
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLConcatenateLayer.cpp | 39 unsigned int axis{ 0 }; member
54 …nateLayer::configure(std::vector<const ICLTensor *> &inputs_vector, ICLTensor *output, size_t axis) in configure()
59 …xt &compile_context, std::vector<const ICLTensor *> &inputs_vector, ICLTensor *output, size_t axis) in configure()
79 …date(const std::vector<const ITensorInfo *> &inputs_vector, const ITensorInfo *output, size_t axis) in validate()
|
| D | CLReverse.cpp | 33 void CLReverse::configure(const ICLTensor *input, ICLTensor *output, const ICLTensor *axis) in configure()
38 …LCompileContext &compile_context, const ICLTensor *input, ICLTensor *output, const ICLTensor *axis) in configure()
46 …s CLReverse::validate(const ITensorInfo *input, const ITensorInfo *output, const ITensorInfo *axis) in validate()
|
| D | CLGather.cpp | 33 … CLGather::configure(const ICLTensor *input, const ICLTensor *indices, ICLTensor *output, int axis) in configure()
38 …xt &compile_context, const ICLTensor *input, const ICLTensor *indices, ICLTensor *output, int axis) in configure()
46 …validate(const ITensorInfo *input, const ITensorInfo *indices, const ITensorInfo *output, int axis) in validate()
|
| /external/tensorflow/tensorflow/python/kernel_tests/control_flow/ |
| D | scan_ops_test.py | 29 def numpy_reverse(x, axis): argument
40 def handle_options(func, x, axis, exclusive, reverse): argument
76 def _compare(self, x, axis, exclusive, reverse): argument
83 def _compareAll(self, x, axis): argument
165 def _compareGradient(self, shape, axis, exclusive, reverse): argument
209 def _compare(self, x, axis, exclusive, reverse): argument
216 def _compareAll(self, x, axis): argument
291 def _compareGradient(self, shape, axis, exclusive, reverse): argument
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | reduce_test_util.cc | 35 const std::set<tflite::gpu::Axis> axis{Axis::CHANNELS}; in ReduceSumChannelsIntTest() local
75 const std::set<tflite::gpu::Axis> axis{Axis::CHANNELS}; in ReduceProductChannelsUIntTest() local
116 const std::set<tflite::gpu::Axis> axis{Axis::HEIGHT, Axis::WIDTH}; in MeanHWTest() local
143 const std::set<tflite::gpu::Axis> axis{Axis::CHANNELS}; in ReduceSumChannelsTest() local
174 const std::set<tflite::gpu::Axis> axis{Axis::CHANNELS}; in ReduceProductChannelsTest() local
206 const std::set<tflite::gpu::Axis> axis{Axis::CHANNELS}; in ReduceMaxChannelsTest() local
234 const std::set<tflite::gpu::Axis> axis{Axis::CHANNELS}; in ReduceMinChannelsTest() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | shape.cc | 42 Axis axis; member
54 std::string ToString(Axis axis) { in ToString()
120 int GetAxisIndex(Layout layout, Axis axis) { in GetAxisIndex()
124 bool HasAxis(Layout layout, Axis axis) { in HasAxis()
|
| /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/tests/validation/fixtures/dynamic_fusion/operators/ |
| D | SoftmaxFixture.h | 50 void setup(TensorShape shape, DataType data_type, float beta, size_t axis, bool is_log) in setup()
81 …compute_target(const TensorShape &shape, DataType data_type, float beta, int32_t axis, bool is_log) in compute_target()
127 …pute_reference(const TensorShape &shape, DataType data_type, float beta, int32_t axis, bool is_log) in compute_reference()
147 void setup(TensorShape shape, DataType data_type, float beta, size_t axis, bool is_log) in setup()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | scan_ops_test.py | 30 def numpy_reverse(x, axis): argument
41 def handle_options(func, x, axis, exclusive, reverse): argument
78 def _compare(self, x, axis, exclusive, reverse): argument
87 def _compareAll(self, x, axis): argument
165 def _compare(self, x, axis, exclusive, reverse): argument
174 def _compareAll(self, x, axis): argument
|
| /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()
79 const TfLiteTensor* axis; in Prepare() local
109 const TfLiteTensor* axis; in Eval() local
|