Searched refs:CheckArgs (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | relu_grad.cc | 38 auto dout = CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(prim_name, input_args, 0); in InferShape()
39 auto out = CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(prim_name, input_args, 1); in InferShape()
54 auto dout = CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(prim_name, input_args, 0); in InferType()
55 auto out = CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(prim_name, input_args, 1); in InferType()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | ctcloss.cc | 36 auto inputs = CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(op_name, input_args, 0); in CheckCTCLossInputs()
37 …auto labels_indices = CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(op_name, input_arg… in CheckCTCLossInputs()
38 …auto labels_values = CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(op_name, input_args… in CheckCTCLossInputs()
39 …auto sequence_length = CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(op_name, input_ar… in CheckCTCLossInputs()
|
| D | layer_norm.cc | 50 …auto input_x = CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(op_name, input_args, x_in… in LayerNormInfer()
51 …auto gamma = CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(op_name, input_args, gamma_… in LayerNormInfer()
52 …auto beta = CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(op_name, input_args, beta_in… in LayerNormInfer()
|
| D | gather.cc | 35 CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(op_name, input_args, 0); in GatherInfer()
37 CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(op_name, input_args, 1); in GatherInfer()
|
| D | bias_add.cc | 34 auto x = CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(prim_name, input_args, 0); in InferShape()
35 auto bias = CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(prim_name, input_args, 1); in InferShape()
|
| D | relu.cc | 34 (void)CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(prim_name, input_args, 0); in InferShape()
|
| D | neighborexchange.cc | 112 (void)CheckAndConvertUtils::CheckArgs<abstract::AbstractTuple>(prim_name, input_args, 0); in Check()
|
| /third_party/mindspore/mindspore/ccsrc/common/ |
| D | trans.cc | 1064 bool CheckArgs(const FormatArgs &args, size_t *size, size_t *total_size) { in CheckArgs() function
1169 if (!CheckArgs(args, &size, &total_size)) { in NchwTo4D()
1201 if (!CheckArgs(args, &size, &total_size)) { in ToNchw()
1342 if (!CheckArgs(args, &size, &total_size)) { in NchwToFracZc04()
1657 if (!CheckArgs(args, &size, &total_size)) { in NchwToC1hwncoc0()
1698 if (!CheckArgs(args, &size, &total_size)) { in C1hwncoc0ToNchw()
|
| /third_party/mindspore/mindspore/core/utils/ |
| D | check_convert_utils.h | 266 …static std::shared_ptr<T> CheckArgs(const std::string &op, const AbstractBasePtrList &args_spec_li… in CheckArgs() function
|
| D | check_convert_utils.cc | 420 …auto abstract = CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(prim_name, input_args, L… in GetTensorInputShape()
|