| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| H A D | reduction_ops_test.py | 135 def _tf_reduce(self, x, reduction_axes, keepdims): argument
138 def _np_reduce(self, x, reduction_axes, keepdims): argument
153 def _compare(self, x, reduction_axes, keepdims, feed_dict=None): argument
196 def _tf_reduce(self, x, reduction_axes, keepdims): argument
199 def _np_reduce(self, x, reduction_axes, keepdims): argument
421 def _tf_reduce(self, x, reduction_axes, keepdims): argument
424 def _np_reduce(self, x, reduction_axes, keepdims): argument
543 def _tf_reduce(self, x, reduction_axes, keepdims): argument
546 def _np_reduce(self, x, reduction_axes, keepdims): argument
628 def _tf_reduce(self, x, reduction_axes, keepdims): argument
[all …]
|
| H A D | reduction_ops_test_big.py | 28 def _tf_reduce(self, x, reduction_axes, keepdims): argument
35 def _tf_reduce_max(self, x, reduction_axes, keepdims): argument
38 def _tf_reduce_all(self, x, reduction_axes, keepdims): argument
41 def _tf_reduce_mean(self, x, reduction_axes, keepdims): argument
44 def _tf_reduce_sum(self, x, reduction_axes, keepdims): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | math_ops.py | 2161 def _may_reduce_to_scalar(keepdims, axis, output): argument
2176 keepdims=None, argument
2251 def reduce_sum(input_tensor, axis=None, keepdims=False, name=None): argument
2318 keepdims=False, argument
2329 def reduce_euclidean_norm(input_tensor, axis=None, keepdims=False, name=None): argument
2382 keepdims=None, argument
2455 keepdims=None, argument
2522 keepdims=None, argument
2587 def reduce_mean(input_tensor, axis=None, keepdims=False, name=None): argument
2647 def reduce_variance(input_tensor, axis=None, keepdims=False, name=None): argument
[all …]
|
| H A D | sparse_ops.py | 1231 sp_input, axis=None, keepdims=None, output_is_sparse=False, name=None): argument
1331 def sparse_reduce_max(sp_input, axis=None, keepdims=None, argument
1416 keepdims=None, argument
1465 sp_input, axis=None, keepdims=None, output_is_sparse=False, name=None): argument
1549 def sparse_reduce_sum(sp_input, axis=None, keepdims=None, argument
1617 keepdims=None, argument
|
| H A D | nn_impl.py | 1184 keepdims=None): argument
1258 def sufficient_statistics_v2(x, axes, shift=None, keepdims=False, name=None): argument
1327 keepdims=None): argument
1391 keepdims=False, argument
1425 keepdims=None): argument
1503 def weighted_moments_v2(x, axes, frequency_weights, keepdims=False, name=None): argument
|
| H A D | linalg_ops.py | 562 keepdims=None, argument
636 keepdims=None, argument
|
| H A D | string_ops.py | 313 keepdims=None): argument
334 keepdims=False, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_math_ops.py | 480 keepdims, argument
612 keepdims=None, argument
628 keepdims=None, argument
643 keepdims=None, argument
658 keepdims=None, argument
673 keepdims=None, argument
698 keepdims=False, argument
720 keepdims=False, argument
736 keepdims=None, argument
748 keepdims=None, argument
|
| H A D | strings_reduce_join_op_test.py | 119 keepdims, argument
|
| H A D | ragged_reduce_op_test.py | 632 def testReduce(self, ragged_reduce_op, rt_input, axis, keepdims, expected): argument
|
| H A D | ragged_string_ops.py | 666 keepdims=None, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| H A 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
473 keepdims=None, 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
590 def amax(a, axis=None, out=None, keepdims=None): argument
605 def amin(a, axis=None, out=None, keepdims=None): argument
620 def var(a, axis=None, dtype=None, out=None, ddof=0, keepdims=None): # pylint: disable=missing-docs… argument
629 def reduce_fn(input_tensor, axis, keepdims): argument
[all …]
|
| H A D | __init__.py | 185 def max(a, axis=None, keepdims=None): argument
190 def min(a, axis=None, keepdims=None): argument
|
| H A D | np_utils.py | 667 def reduce_all(input_tensor, axis=None, keepdims=False): argument
676 def reduce_any(input_tensor, axis=None, keepdims=False): argument
|
| H A D | np_math_ops.py | 829 def nan_reduction(a, axis=None, dtype=None, keepdims=False): argument
846 def nanmean(a, axis=None, dtype=None, keepdims=None): # pylint: disable=missing-docstring argument
1129 def ptp(a, axis=None, keepdims=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/ |
| H A D | backend.py | 2282 def max(x, axis=None, keepdims=False): argument
2302 def min(x, axis=None, keepdims=False): argument
2322 def sum(x, axis=None, keepdims=False): argument
2342 def prod(x, axis=None, keepdims=False): argument
2393 def var(x, axis=None, keepdims=False): argument
2415 def std(x, axis=None, keepdims=False): argument
2443 def mean(x, axis=None, keepdims=False): argument
2465 def any(x, axis=None, keepdims=False): argument
2483 def all(x, axis=None, keepdims=False): argument
2610 def logsumexp(x, axis=None, keepdims=False): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/layers/ |
| H A D | pooling.py | 888 def __init__(self, data_format='channels_last', keepdims=False, **kwargs): argument
1055 def __init__(self, data_format=None, keepdims=False, **kwargs): argument
1193 def __init__(self, data_format=None, keepdims=False, **kwargs): argument
|
| /aosp_15_r20/external/pytorch/test/torch_np/ |
| H A D | test_reductions.py | 411 def _check_out_axis(self, axis, dtype, keepdims): argument
436 def test_out_axis(self, func, axis, dtype, keepdims): argument
|
| /aosp_15_r20/external/pytorch/torch/_numpy/ |
| H A D | _reductions_impl.py | 328 keepdims=False, argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | constant_fold.cpp | 465 bool keepdims = in runTorchBackendForOnnx() local
|
| H A D | shape_type_inference.cpp | 941 int64_t keepdims = 1; in ProcessReduceNode() local
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | lowering.py | 5081 def _make_reduction_inner(x, *, axis, keepdims, dtype, override_return_dtype): argument
5131 def inner(x, axis=None, keepdims=False, *, dtype=None): argument
5558 def sum_(x, axis=None, keepdims=False, *, dtype=None): argument
5706 def prod(x, axis=None, keepdims=False, *, dtype=None): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_reductions.py | 1229 def _amin_wrapper(x, dim=None, keepdims=False): argument
1232 def _amax_wrapper(x, dim=None, keepdims=False): argument
|