| /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
242 def _ascending_sort(values, axis, return_argsort=False): argument
|
| D | math_ops.py | 121 def linspace_nd(start, stop, num, name=None, axis=0): argument
262 axis=None, argument
273 def argmax_v2(input, axis=None, output_type=dtypes.int64, name=None): argument
316 axis=None, argument
327 def argmin_v2(input, axis=None, output_type=dtypes.int64, name=None): argument
1905 def _ReductionDims(x, axis): # pylint: disable=invalid-name argument
1929 def _may_reduce_to_scalar(keepdims, axis, output): argument
1943 axis=None, argument
2019 def reduce_sum(input_tensor, axis=None, keepdims=False, name=None): argument
2085 axis=None, argument
[all …]
|
| D | partitioned_variables.py | 72 def variable_axis_size_partitioner( argument
158 def min_max_variable_partitioner(max_partitions=1, axis=0, argument
222 def fixed_size_partitioner(num_shards, axis=0): argument
|
| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_array_ops.py | 32 def expand_dims(input, axis=None, name=None, dim=None): # pylint: disable=redefined-builtin argument
65 def expand_dims_v2(input, axis, name=None): # pylint: disable=redefined-builtin argument
95 def _expand_dims_impl(st, axis, name=None): # pylint: disable=redefined-builtin argument
139 def _expand_st_row_partitions(st, axis): 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_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
|
| D | ragged_math_ops.py | 437 axis, argument
567 def reduce_sum(input_tensor, axis=None, keepdims=None, name=None): argument
579 def reduce_prod(input_tensor, axis=None, keepdims=None, name=None): argument
590 def reduce_min(input_tensor, axis=None, keepdims=None, name=None): argument
601 def reduce_max(input_tensor, axis=None, keepdims=None, name=None): argument
612 def reduce_mean(input_tensor, axis=None, keepdims=None, name=None): argument
638 def reduce_all(input_tensor, axis=None, keepdims=None, name=None): argument
646 def reduce_any(input_tensor, axis=None, keepdims=None, name=None): argument
|
| D | ragged_util_test.py | 92 def testRepeat(self, data, repeats, expected, axis=None): argument
139 def testValuesMatchesNumpy(self, mode, data, repeats, axis): argument
204 axis, argument
|
| D | ragged_concat_ops.py | 33 def concat(values, axis, name=None): argument
75 def stack(values, axis=0, name=None): argument
124 def _ragged_stack_concat_helper(rt_inputs, axis, stack_values): argument
|
| D | ragged_one_hot_op_test.py | 90 axis=None, argument
123 axis=None, argument
157 axis=None, argument
|
| D | ragged_concat_op_test.py | 231 axis, argument
273 axis, argument
289 def testRuntimeError(self, rt_inputs, axis, error, message, argument
|
| D | ragged_row_lengths_op_test.py | 117 axis=1, argument
139 def testErrors(self, rt_input, exception, message=None, axis=1): 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, keepdims=None): argument
[all …]
|
| D | __init__.py | 188 def max(a, axis=None, keepdims=None): argument
193 def min(a, axis=None, keepdims=None): argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce_test.cc | 84 std::initializer_list<int> axis, bool keep_dims) { in MeanOpConstModel()
98 const TensorData& axis, bool keep_dims) { in MeanOpDynamicModel()
113 std::initializer_list<int> axis, bool keep_dims) { in SumOpConstModel()
127 const TensorData& axis, bool keep_dims) { in SumOpDynamicModel()
142 std::initializer_list<int> axis, bool keep_dims) { in ProdOpConstModel()
156 const TensorData& axis, bool keep_dims) { in ProdOpDynamicModel()
171 std::initializer_list<int> axis, bool keep_dims) { in MaxOpConstModel()
185 const TensorData& axis, bool keep_dims) { in MaxOpDynamicModel()
200 std::initializer_list<int> axis, bool keep_dims) { in MinOpConstModel()
214 const TensorData& axis, bool keep_dims) { in MinOpDynamicModel()
[all …]
|
| D | expand_dims.cc | 34 int axis, TfLiteTensor* output) { in ExpandTensorDim()
56 const TfLiteTensor& axis, int* axis_value) { in GetAxisValueFromTensor()
77 const TfLiteTensor* axis; in Prepare() local
98 const TfLiteTensor* axis; in Eval() local
|
| /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/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/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/python/keras/layers/preprocessing/ |
| D | normalization_test.py | 145 def test_layer_computation(self, adapt_data, axis, test_data, use_dataset, argument
231 def test_zeros_fail_init(self, axis): argument
245 def test_bad_axis_fail_build(self, axis): argument
258 def test_axis_permutations(self, axis): argument
|
| /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
|
| /external/s2-geometry-library-java/src/com/google/common/geometry/ |
| D | S2Cap.java | 42 private final S2Point axis; field in S2Cap
52 private S2Cap(S2Point axis, double height) { in S2Cap()
63 public static S2Cap fromAxisHeight(S2Point axis, double height) { in fromAxisHeight()
73 public static S2Cap fromAxisAngle(S2Point axis, S1Angle angle) { in fromAxisAngle()
88 public static S2Cap fromAxisArea(S2Point axis, double area) { in fromAxisArea()
105 public S2Point axis() { in axis() method in S2Cap
|
| /external/tensorflow/tensorflow/python/ops/signal/ |
| D | dct_ops.py | 32 def _validate_dct_arguments(input_tensor, dct_type, n, axis, norm): argument
55 def dct(input, type=2, n=None, axis=-1, norm=None, name=None): # pylint: disable=redefined-builtin argument
187 def idct(input, type=2, n=None, axis=-1, norm=None, name=None): # pylint: disable=redefined-builtin argument
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | pack_test.cc | 51 const float* input2_data, int axis, in TestPackTwoInputsFloat()
84 const int* input3_dims_data, const float* input3_data, int axis, in TestPackThreeInputsFloat()
118 const uint8_t* input2_data, int axis, in TestPackTwoInputsQuantized()
155 const int32_t* input2_data, int axis, in TestPackTwoInputsQuantized32()
197 const int axis = 0; in TF_LITE_MICRO_TEST() local
213 const int axis = 1; in TF_LITE_MICRO_TEST() local
229 const int axis = -1; in TF_LITE_MICRO_TEST() local
244 const int axis = 1; in TF_LITE_MICRO_TEST() local
259 const int axis = 1; in TF_LITE_MICRO_TEST() local
274 const int axis = 1; in TF_LITE_MICRO_TEST() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/ |
| D | resolve_constant_concatenation_test.cc | 109 void PrepareModel(Model* model, int axis) { in PrepareModel()
165 const int axis = 0; in TEST_F() local
187 const int axis = 1; in TEST_F() local
209 const int axis = 2; in TEST_F() local
|