| /third_party/mindspore/mindspore-src/source/tests/st/ops/ |
| D | test_ops_argmax_ext.py | 28 def generate_expect_forward_output(x, dim=None, keepdim=False): argument
32 def generate_expect_backward_output(x, dim=None, keepdim=False): argument
37 def argmax_ext_forward_func(x, dim=None, keepdim=False): argument
42 def argmax_ext_backward_func(x, dim=None, keepdim=False): argument
|
| D | test_func_norm.py | 25 def construct(self, a, norm_ord=None, dim=None, keepdim=False, dtype=None): argument
|
| D | test_func_sum.py | 26 def construct(self, x, dim=None, keepdim=False, dtype=None): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/opapi/aclnn/ |
| D | norm_aclnn_kernel.cc | 45 auto keepdim = transform::ConvertKernelTensor<bool>(inputs[kIndex3]); in GetWorkSpaceInfo() local
54 auto keepdim = transform::ConvertKernelTensor<bool>(inputs[kIndex3]); in Launch() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/extend/ |
| D | array_func.py | 80 def max(input, dim=None, keepdim=False): argument
129 def min(input, dim=None, keepdim=False): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/mint/ |
| D | __init__.py | 312 def any(input, dim=None, keepdim=False): argument
372 def all(input, dim=None, keepdim=False): argument
490 def mean(input, dim=None, keepdim=False, *, dtype=None): argument
579 def prod(input, dim=None, keepdim=False, *, dtype=None): argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/ |
| D | test_ops_argmax_ext.cc | 31 ValuePtr keepdim; member
44 auto keepdim = param.keepdim->ToAbstract(); in TEST_P() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | argmax_ext.cc | 45 auto keepdim = GetScalarValue<bool>(input_args[kInputIndex2]->GetValue()); in InferShape() local
75 const auto &keepdim = GetScalarValue<bool>(input_values[kInputIndex2]); in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/ |
| D | nuclear_norm_cpu_kernel.h | 91 bool keepdim = false; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/ |
| D | argmax_ext.cc | 29 … const std::optional<Int64ImmPtr> &dim, const BoolImmPtr &keepdim) { in ArgMaxAscendCustomize()
|
| D | norm.cc | 32 const BoolImmPtr &keepdim, const std::optional<Int64ImmPtr> &dtype) { in NormAscendCustomize()
|
| /third_party/mindspore/mindspore-src/source/tests/st/tensor/ |
| D | test_norm.py | 25 def construct(self, a, norm_ord=None, dim=None, keepdim=False, dtype=None): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | nuclear_norm.cc | 73 auto keepdim = GetValue<bool>(primitive->GetAttr("keepdim")); in NuclearNormInferShape() local
|
| D | quantile.cc | 155 void Quantile::set_keepdim(bool keepdim) { (void)AddAttr(kKeepdim, api::MakeValue(keepdim)); } in set_keepdim()
|
| D | median.cc | 107 bool keepdim = GetValue<bool>(check_keepdim); in MedianInferShape() local
|
| D | op_utils.cc | 448 auto keepdim = GetValue<bool>(input_args[2]->GetValue()); in ReduceExtInferShape() local
|
| /third_party/mindspore/mindspore-src/source/config/ |
| D | super_bar_config.json | 175 "keepdim": "keep_dims" string
251 "keepdim": "keep_dims" string
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/ |
| D | math_func.py | 7011 def norm(A, ord=None, dim=None, keepdim=False, *, dtype=None): argument
7279 def norm_ext(A, ord=None, dim=None, keepdim=False, *, dtype=None): argument
10087 def sum(input, dim=None, keepdim=False, *, dtype=None): argument
|
| D | array_func.py | 5366 def argmax(input, dim=None, keepdim=False): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | nn_ops.py | 9187 def __init__(self, dim=None, keepdim=False): argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/ops/ |
| D | test_ops.py | 1493 def __init__(self, dim=None, keepdim=False): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/_extends/parse/ |
| D | standard_method.py | 3344 def norm(A, ord=None, dim=None, keepdim=False, *, dtype=None): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/common/ |
| D | tensor.py | 1688 def norm(self, ord=None, dim=None, keepdim=False, *, dtype=None): argument
|