Home
last modified time | relevance | path

Searched defs:keepdims (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/numpy/
Dmath_ops.py107 def count_nonzero(x, axis=None, keepdims=False): argument
582 def mean(a, axis=None, keepdims=False, dtype=None): argument
873 def std(x, axis=None, ddof=0, keepdims=False): argument
916 def var(x, axis=None, ddof=0, keepdims=False): argument
957 def ptp(x, axis=None, keepdims=False): argument
1382 def amax(a, axis=None, keepdims=False, initial=None, where=True): argument
1437 def amin(a, axis=None, keepdims=False, initial=None, where=True): argument
2427 def _reduce(a, reduce_fn, cmp_fn=None, axis=None, keepdims=False, initial=None, where=True, dtype=N… argument
2474 def nanmax(a, axis=None, dtype=None, keepdims=False): argument
2522 def nanmin(a, axis=None, dtype=None, keepdims=False): argument
[all …]
Dlogic_ops.py863 def sometrue(a, axis=None, keepdims=False): argument
/third_party/mindspore/mindspore/_extends/parse/
Dstandard_method.py590 def max(x, axis=None, keepdims=False, initial=None, where=True): # pylint: disable=redefined-builtin argument
635 def min(x, axis=None, keepdims=False, initial=None, where=True): # pylint: disable=redefined-builtin argument
1077 def ptp(x, axis=None, keepdims=False): argument
1176 def var(x, axis=None, ddof=0, keepdims=False): argument
1233 def std(x, axis=None, ddof=0, keepdims=False): argument
1271 def sum(x, axis=None, dtype=None, keepdims=False, initial=None): # pylint: disable=redefined-builtin argument
/third_party/mindspore/mindspore/common/
Dtensor.py966 def max(self, axis=None, keepdims=False, initial=None, where=True): argument
1011 def min(self, axis=None, keepdims=False, initial=None, where=True): argument
1094 def ptp(self, axis=None, keepdims=False): argument
1667 def var(self, axis=None, ddof=0, keepdims=False): argument
1724 def std(self, axis=None, ddof=0, keepdims=False): argument
1762 def sum(self, axis=None, dtype=None, keepdims=False, initial=None): argument
/third_party/mindspore/tests/st/ops/gpu/
Dtest_reduce_mean_op.py270 def __init__(self, x, axis, keepdims=False): argument
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_generate_strategy.cc805 bool keepdims = false; in GetKeepDims() local
899 bool keepdims = GetKeepDims(ops, incoming_op_index); in ModifyStrategyIfArgIncoming() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/model/
Dop_node.cc351 auto keepdims = GetValue<bool>(attrs.find("keep_dims")->second); in InferShape() local
/third_party/mindspore/mindspore/ops/composite/multitype_ops/
D_compile_utils.py945 def reduce_(a, reduce_fn, cmp_fn=None, axis=None, keepdims=False, initial=None, where=True, dtype=N… argument