| /third_party/mindspore/mindspore/ |
| D | _checkparam.py | 99 def _check_3d_int_or_tuple(arg_name, arg_value, prim_name, allow_five=False, ret_five=False, argument
147 def check_number(arg_value, value, rel, arg_type=int, arg_name=None, prim_name=None): argument
174 def check_is_number(arg_value, arg_type, arg_name=None, prim_name=None): argument
192 …_number_range(arg_value, lower_limit, upper_limit, rel, value_type, arg_name=None, prim_name=None): argument
218 …def check(arg_name, arg_value, value_name, value, rel=Rel.EQ, prim_name=None, excp_cls=ValueError): argument
231 def check_int(arg_value, value, rel, arg_name=None, prim_name=None): argument
241 def check_is_int(arg_value, arg_name=None, prim_name=None): argument
253 def check_equal_int(arg_value, value, arg_name=None, prim_name=None): argument
263 def check_positive_int(arg_value, arg_name=None, prim_name=None): argument
274 def check_negative_int(arg_value, arg_name=None, prim_name=None): argument
[all …]
|
| /third_party/mindspore/mindspore/ops/composite/ |
| D | math_ops.py | 122 def _check_axes(axes, prim_name=None): argument
142 def _typecheck_input(x1_type, x2_type, prim_name=None): argument
155 def _axes_int_check(x1_shape, x2_shape, axes, prim_name=None): argument
177 def _validate_axes(x1_shape, x2_shape, axes, prim_name=None): argument
235 def tensor_dot(x1, x2, axes, prim_name='tensor_dot'): argument
306 def _check_invalid_input(x1_shape, x2_shape, prim_name=None): argument
314 def _typecheck_input_dot(x1_type, x2_type, prim_name=None): argument
333 def dot(x1, x2, prim_name=None): argument
422 def _get_batch_size(x1_shape, x2_shape, prim_name=None): argument
434 def _typecheck_input_batch_dot(x1_type, x2_type, prim_name=None): argument
[all …]
|
| D | array_ops.py | 119 def _check_sequence_mask_input_len(input_shape, prim_name=None): argument
129 def sequence_mask(lengths, maxlen=None, prim_name='sequence_mask'): argument
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | timedistributed.py | 27 def _check_reshape_pos(reshape_pos, inputs_shape, outputs_shape, prim_name=None): argument
38 def _check_expand_dims_axis(time_axis, ndim, prim_name=None): argument
53 def _check_data(flag, prim_name=None): argument
60 def _check_inputs_dim(shape, prim_name=None): argument
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | gather_d.cc | 30 auto prim_name = primitive->name(); in GatherDInferShape() local
55 auto prim_name = prim->name(); in GatherDInferType() local
65 auto prim_name = primitive->name(); in GatherDInfer() local
|
| D | relu.cc | 32 auto prim_name = primitive->name(); in InferShape() local
43 auto prim_name = prim->name(); in InferType() local
|
| D | zeros.cc | 31 auto prim_name = primitive->name(); in ZerosInferShape() local
41 auto prim_name = prim->name(); in ZerosInferType() local
|
| D | dynamic_broadcast_gradient_args.cc | 30 int64_t CheckInputsAndGetShape(const AbstractBasePtr &input_arg, const string &prim_name) { in CheckInputsAndGetShape()
57 auto prim_name = primitive->name(); in Infer() local
|
| D | ones.cc | 30 auto prim_name = primitive->name(); in OnesInferShape() local
39 auto prim_name = prim->name(); in OnesInferType() local
|
| D | addn.cc | 67 auto prim_name = prim->name(); in AddNInferType() local
94 auto prim_name = primitive->name(); in AddNInfer() local
|
| D | rank.h | 34 Rank() : PrimitiveC(kNameRank) { auto prim_name = name(); } in Rank() local
|
| D | reluv2.cc | 58 auto prim_name = primitive->name(); in InferShape() local
89 auto prim_name = prim->name(); in InferType() local
|
| D | tile.cc | 54 auto prim_name = primitive->name(); in TileInferShape() local
78 auto prim_name = prim->name(); in TileInferType() local
|
| D | sparse_apply_r_m_s_prop.cc | 31 auto prim_name = primitive->name(); in InferShape() local
74 auto prim_name = prim->name(); in InferType() local
|
| D | fft_real.cc | 30 auto prim_name = primitive->name(); in FftRealInfer() local
|
| D | sparse_to_dense.cc | 30 auto prim_name = primitive->name(); in SparseToDenseInfer() local
|
| D | bias_add.cc | 33 auto prim_name = primitive->name(); in InferShape() local
78 auto prim_name = prim->name(); in InferType() local
|
| D | add.cc | 30 auto prim_name = primitive->name(); in AddInfer() local
|
| /third_party/mindspore/mindspore/core/utils/ |
| D | check_convert_utils.cc | 311 const std::string &prim_name) { in CheckPositiveVector()
327 … const std::set<std::string> &check_list, const std::string &prim_name) { in CheckString()
348 int64_t match_value, const std::string &prim_name) { in CheckInteger()
372 const std::string &prim_name) { in CheckInputArgs()
382 const std::string &prim_name) { in GetInputTensorType()
417 abstract::ShapePtr CheckAndConvertUtils::GetTensorInputShape(const std::string &prim_name, in GetTensorInputShape()
433 int64_t value, const string &prim_name, ExceptionType) { in Check()
456 … const std::set<TypePtr> &check_list, const std::string &prim_name) { in CheckTensorTypeSame()
495 … const std::set<TypePtr> &check_list, const std::string &prim_name) { in CheckTensorTypeValid()
516 const std::string &prim_name) { in CheckTensorIntValue()
[all …]
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | relu_grad_v2.cc | 38 auto prim_name = prim->name(); in InferType() local
51 auto prim_name = primitive->name(); in ReLUGradV2Infer() local
|
| D | relu_grad.cc | 32 auto prim_name = primitive->name(); in InferShape() local
50 auto prim_name = prim->name(); in InferType() local
|
| D | bias_add_grad.cc | 42 auto prim_name = primitive->name(); in InferShape() local
62 auto prim_name = prim->name(); in InferType() local
|
| D | conv2d_backprop_filter.cc | 34 auto prim_name = primitive->name(); in TransStrideTo4D() local
55 auto prim_name = primitive->name(); in Conv2DBackpropFilterInferShape() local
113 auto prim_name = prim->name(); in Conv2DBackpropFilterInferType() local
233 auto prim_name = primitive->name(); in Conv2DBackpropFilterInfer() local
|
| D | conv2d_backprop_input.cc | 35 auto prim_name = primitive->name(); in SetPadList() local
85 auto prim_name = primitive->name(); in Conv2DBackpropInputInferShape() local
147 auto prim_name = prim->name(); in Conv2DBackpropInputInferType() local
160 auto prim_name = primitive->name(); in Conv2DBackpropInputInfer() local
|
| /third_party/mindspore/mindspore/ops/_utils/ |
| D | utils.py | 25 def get_broadcast_shape(x_shape, y_shape, prim_name, shape_type=""): argument
78 def get_concat_offset(x_shp, x_type, axis, prim_name): argument
|