| /third_party/mindspore/mindspore/core/ops/ |
| D | split_v.cc | 35 …CheckAndConvertUtils::CheckInRange("split_dim", split_dim, kIncludeLeft, {-x_rank, x_rank}, prim_n… in InferShape()
49 …CheckAndConvertUtils::CheckInRange("elements of size_splits", size_splits[LongToSize(i)], kInclude… in InferShape()
63 …CheckAndConvertUtils::CheckInRange("elements of size_splits", size_splits[LongToSize(i)], kInclude… in InferShape()
|
| D | softmax.cc | 42 CheckAndConvertUtils::CheckInRange<int64_t>("axis", item, kIncludeLeft, {-rank, rank}, op_name); in Init()
63 CheckAndConvertUtils::CheckInRange<int64_t>("axis", item, kIncludeLeft, {-rank, rank}, op_name); in SoftMaxInferShape()
|
| D | control_depend.cc | 24 …CheckAndConvertUtils::CheckInRange<int64_t>(kDependMode, depend_mode, kIncludeBoth, {0, 1}, name()… in set_depend_mode()
|
| D | batch_norm.cc | 37 …CheckAndConvertUtils::CheckInRange<float>(kEpsilon, epsilon, kIncludeBoth, {0.0, 1.0}, this->name(… in set_epsilon()
47 …CheckAndConvertUtils::CheckInRange<float>(kMomentum, momentun, kIncludeBoth, {0.0, 1.0}, this->nam… in set_momentum()
|
| D | expand_dims.cc | 43 …CheckAndConvertUtils::CheckInRange<int64_t>("axis", dim_val, kIncludeBoth, {-SizeToLong(rank) - 1,… in ExpandDimsInfer()
|
| D | roll.cc | 37 …CheckAndConvertUtils::CheckInRange("axis value", axis, kIncludeLeft, {-x_rank, x_rank}, prim_name); in InferShape()
|
| D | dropout.cc | 30 …CheckAndConvertUtils::CheckInRange<float>(kKeepProb, keep_prob, kIncludeRight, {0.0, 1.0}, this->n… in set_keep_prob()
|
| D | l2_normalize.cc | 53 …CheckAndConvertUtils::CheckInRange<int64_t>("axis", axis, kIncludeLeft, {-x_rank, x_rank}, prim_na… in L2NormalizeInfer()
|
| D | arg_min.cc | 48 …CheckAndConvertUtils::CheckInRange<int64_t>("axis", axis, kIncludeLeft, {-x_rank, x_rank}, prim_na… in ArgMinInfer()
|
| D | cdist.cc | 38 CheckAndConvertUtils::CheckInRange("input_x dim", x_size, kIncludeBoth, {2, 3}, "Cdist"); in InferShape()
|
| D | log_softmax.cc | 50 CheckAndConvertUtils::CheckInRange<int64_t>("axis", axis, kIncludeLeft, {-rank, rank}, op_name); in LogSoftmaxInferShape()
|
| D | arg_max.cc | 28 …CheckAndConvertUtils::CheckInRange<int64_t>("argmax axis", axis, kIncludeLeft, {-x_rank, x_rank}, … in InferShape()
|
| D | reduce.cc | 30 CheckAndConvertUtils::CheckInRange("axis", one_axis, kIncludeLeft, {-dim, dim}, "Reduce"); in reduce_one_axis()
|
| D | squeeze.cc | 39 …CheckAndConvertUtils::CheckInRange<int64_t>("axis_or_elememt", item, kIncludeBoth, {-len, len + 1}… in InferShape()
|
| D | concat.cc | 51 CheckAndConvertUtils::CheckInRange<int64_t>("Concat axis", axis_temp, kIncludeBoth, in ConcatInfer()
|
| D | binary_cross_entropy.cc | 36 …CheckAndConvertUtils::CheckInRange("binary_cross_entropy_infer", input_args.size(), kIncludeBoth, … in BinaryCrossEntroyInferShape()
|
| D | index_add.cc | 40 …CheckAndConvertUtils::CheckInRange("axis", axis, kIncludeNeither, {-x_rank - 1, x_rank}, prim_name… in IndexAddInferShape()
|
| D | one_hot.cc | 38 …CheckAndConvertUtils::CheckInRange<int64_t>("axis", axis, kIncludeBoth, {-1, SizeToLong(in_shape.s… in OneHotInferShape()
|
| D | gather.cc | 68 …CheckAndConvertUtils::CheckInRange<int64_t>("axis", axis_val, kIncludeLeft, {-params_rank, params_… in GatherInfer()
|
| D | bias_add.cc | 46 …CheckAndConvertUtils::CheckInRange("dims of input_x", input_shape.size(), kIncludeBoth, {x_min_ran… in InferShape()
|
| D | detection_post_process.cc | 138 …CheckAndConvertUtils::CheckInRange("scores_shape[2]", scores_shape[2], kIncludeBoth, {num_classes,… in DetectionPostProcessInfer()
|
| D | lstm.cc | 127 …CheckAndConvertUtils::CheckInRange<float>(kDropout, dropout, kIncludeBoth, {0.0, 1.0}, this->name(… in set_dropout()
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | dropout_grad.cc | 30 …CheckAndConvertUtils::CheckInRange<float>(kKeepProb, keep_prob, kIncludeRight, {0.0, 1.0}, this->n… in set_keep_prob()
|
| D | cdist_grad.cc | 42 CheckAndConvertUtils::CheckInRange("input_x dim", x_size, kIncludeBoth, {2, 3}, "CdistGrad"); in InferShape()
|
| D | lstm_grad.cc | 55 …CheckAndConvertUtils::CheckInRange<float>(kDropout, dropout, kIncludeBoth, {0.0, 1.0}, this->name(… in set_dropout()
|