| /third_party/mindspore/mindspore/core/ops/fusion/ |
| D | arg_max_fusion.cc | 21 void ArgMaxFusion::Init(const bool keep_dims, const bool out_max_value, const int64_t top_k, const … in Init()
28 void ArgMaxFusion::set_keep_dims(const bool keep_dims) { (void)this->AddAttr(kKeepDims, MakeValue(k… in set_keep_dims()
35 auto keep_dims = GetAttr(kKeepDims); in get_keep_dims() local
|
| D | arg_min_fusion.cc | 21 void ArgMinFusion::Init(bool keep_dims, bool out_max_value, int64_t top_k, int64_t axis) { in Init()
28 void ArgMinFusion::set_keep_dims(const bool keep_dims) { (void)this->AddAttr(kKeepDims, MakeValue(k… in set_keep_dims()
33 auto keep_dims = GetAttr(kKeepDims); in get_keep_dims() local
|
| D | reduce_fusion.cc | 29 void ReduceFusion::set_keep_dims(const bool keep_dims) { (void)this->AddAttr(kKeepDims, MakeValue(k… in set_keep_dims()
66 void ReduceFusion::Init(const bool keep_dims, const ReduceMode mode, const bool reduce_to_end, cons… in Init()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | reduce.cc | 37 const bool keep_dims) { in infer_shape_reduce()
75 auto keep_dims = GetValue<bool>(primitive->GetAttr(kKeepDims)); in InferShape() local
90 void Reduce::set_keep_dims(const bool keep_dims) { (void)this->AddAttr(kKeepDims, MakeValue(keep_di… in set_keep_dims()
94 void Reduce::Init(const bool keep_dims) { this->set_keep_dims(keep_dims); } in Init()
|
| D | all.cc | 23 void All::Init(const int64_t keep_dims) { this->set_keep_dims(keep_dims); } in Init()
25 void All::set_keep_dims(const int64_t keep_dims) { (void)this->AddAttr(kKeepDims, MakeValue(keep_di… in set_keep_dims()
|
| D | reduce_sum.cc | 94 bool keep_dims = GetValue<bool>(keep_dimis_value_ptr); in InferShape() local
|
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_reduce_max.py | 25 def __init__(self, keep_dims): argument
33 def get_output(x, axis, keep_dims, enable_graph_kernel=False): argument
|
| D | test_reduce_min.py | 25 def __init__(self, keep_dims): argument
33 def get_output(x, axis, keep_dims, enable_graph_kernel=False): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | reduce_infer.c | 20 …xes(const TensorC *input, TensorC *output, int *out_shape, size_t out_shape_size, bool keep_dims) { in ReduceOnAllAxes()
32 size_t out_shape_size, bool keep_dims) { in ReduceOnSelectedAxes()
83 bool keep_dims = param->keep_dims_; in ReduceInferShape() local
|
| D | mean_infer.c | 36 bool keep_dims = (bool)(param->keep_dims_); in MeanInferShape() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | reduce_min_fission.cc | 83 bool keep_dims) { in GetInferShape()
129 auto keep_dims = AnfAlgo::GetNodeAttr<bool>(cnode, kAttrKeepDims); in Process() local
|
| /third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/ |
| D | test_ReduceMean.py | 27 def __init__(self, keep_dims, axis): argument
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_argminwithvalue_op.py | 28 def __init__(self, axis=0, keep_dims=False): argument
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | reduce_eliminate.h | 56 auto keep_dims = prim->GetAttr("keep_dims"); in operator() local
|
| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/op/ |
| D | reduce_tensorrt.cc | 58 bool keep_dims = reduce_op->keep_dims(); in AddInnerOp() local
|
| /third_party/mindspore/tests/st/pynative/ |
| D | test_pynative_layernorm_input_and_argmaxwithvalue.py | 149 def __init__(self, axis, keep_dims): argument
162 def __init__(self, input_shape, axis, keep_dims, dtype=np.float32): argument
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_math_ops_check.py | 53 def __init__(self, op_class, keep_dims, axis): argument
|
| /third_party/mindspore/mindspore/ops/composite/ |
| D | math_ops.py | 39 def _check_validate_keepdims(keep_dims, name): argument
44 def count_nonzero(x, axis=(), keep_dims=False, dtype=mstype.int32): argument
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | math.py | 99 def __init__(self, axis, keep_dims=False): argument
983 def __init__(self, axis=None, keep_dims=None): argument
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | reduce_tests.cc | 26 OpParameter *CreateParameter(const std::vector<int> &axis, schema::ReduceMode mode, bool keep_dims)… in CreateParameter()
|
| /third_party/mindspore/mindspore/_extends/parse/ |
| D | standard_method.py | 55 def mean(x, axis=(), keep_dims=False): argument
84 def all_(x, axis=(), keep_dims=False): argument
103 def any_(x, axis=(), keep_dims=False): argument
|
| /third_party/mindspore/tests/vm_impl/ |
| D | vm_me.py | 843 def all_(x, axis=(), keep_dims=False): argument
859 def any_(x, axis=(), keep_dims=False): argument
|
| /third_party/mindspore/mindspore/common/ |
| D | tensor.py | 429 def all(self, axis=(), keep_dims=False): argument
458 def any(self, axis=(), keep_dims=False): argument
540 def mean(self, axis=(), keep_dims=False): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | arithmetic_simplify.cc | 378 bool keep_dims = GetValue<bool>(origin_root->attrs().find("keep_dims")->second); in SetAttributes() local
418 bool keep_dims = GetValue<bool>(origin_root->attrs().find("keep_dims")->second); in SetAttributes() local
|
| /third_party/mindspore/tests/ut/cpp/parallel/ops_info/ |
| D | reduce_method_test.cc | 63 ValuePtr keep_dims = MakeValue(false); in SetUp() local
|