Home
last modified time | relevance | path

Searched refs:CheckInputArgs (Results 1 – 25 of 35) sorted by relevance

12

/third_party/mindspore/mindspore/core/ops/
Dhshrink.cc31 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, 1L, primitive->name()); in InferShape()
37 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, 1, primitive->name()); in InferType()
Dceil.cc33 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, input_num, primitive->name()); in CeilInfer()
Dfft_imag.cc38 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, input_num, primitive->name()); in FftImagInfer()
Dcos.cc41 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, input_num, primitive->name()); in CosInfer()
Dsquared_difference.cc46 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, input_num, primitive->name()); in SquaredDifferenceInfer()
Dfloor.cc45 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, input_num, primitive->name()); in FloorInfer()
Dlogical_or.cc49 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, input_num, primitive->name()); in LogicalOrInfer()
Dzeros_like.cc49 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, input_num, primitive->name()); in ZerosLikeInfer()
Dhashtable_lookup.cc30 CheckAndConvertUtils::CheckInputArgs(input_args, kGreaterEqual, input_num, op_name); in HashtableLookupInfer()
Drsqrt.cc34 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, input_num, prim_name); in InferShape()
Dsoftmax_cross_entropy_with_logits.cc34 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, input_num, prim_name); in SoftmaxCrossEntropyWithLogitsInfer()
Dcustom_normalize.cc54 CheckAndConvertUtils::CheckInputArgs(input_args, kGreaterEqual, input_num, primitive->name()); in CustomNormalizeInfer()
Dsmooth_l1_loss.cc41 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, input_num, prim_name); in SmoothL1LossInfer()
Dquant_dtype_cast.cc37 CheckAndConvertUtils::CheckInputArgs(input_args, kGreaterEqual, input_num, primitive->name()); in QuantDTypeCastInfer()
Dtensor_list_from_tensor.cc29 CheckAndConvertUtils::CheckInputArgs(input_args, kGreaterEqual, input_num, op_name); in TensorListFromTensorInferShape()
Dprelu.cc54 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, input_num, primitive->name()); in PReLUInfer()
Dcdist.cc62 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, input_num, primitive->name()); in CdistInfer()
Dlsh_projection.cc38 CheckAndConvertUtils::CheckInputArgs(input_args, kGreaterEqual, input_num, op_name); in LshProjectionInfer()
Dones.cc55 CheckAndConvertUtils::CheckInputArgs(input_args, kGreaterEqual, input_num, op_name); in OnesInfer()
Dmasked_fill.cc31 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, input_num, op_name); in InferShape()
Dadam.cc28 CheckAndConvertUtils::CheckInputArgs(input_args, kGreaterEqual, input_num, prim_name); in AdamInfer()
Dop_utils.cc60 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, input_num, op_name); in BroadCastInferShape()
/third_party/mindspore/mindspore/core/ops/grad/
Ddropout_grad.cc45 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, input_num, op_name); in DropoutGradInfer()
Dhsigmoid_grad.cc51 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, input_num, prim->name()); in InferType()
Dcdist_grad.cc65 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, input_num, primitive->name()); in CdistGradInfer()

12