/third_party/mindspore/mindspore/core/ops/ |
D | shape.cc | 40 …(void)CheckAndConvertUtils::CheckSubClass("shape type", input_args[0]->BuildType(), valid_params_t… in ShapeInfer() 56 …(void)CheckAndConvertUtils::CheckSubClass("shape type", input_args[0]->BuildType(), valid_params_t… in ShapeInferValue()
|
D | expand_dims.cc | 54 (void)CheckAndConvertUtils::CheckSubClass("x_type", x_type, valid_x_type, prim_name); in ExpandDimsInfer()
|
D | select.cc | 75 (void)CheckAndConvertUtils::CheckSubClass("x_type", x_type, {kTensorType}, prim_name); in SelectInferType() 76 (void)CheckAndConvertUtils::CheckSubClass("y_type", y_type, {kTensorType}, prim_name); in SelectInferType()
|
D | flatten.cc | 45 …(void)CheckAndConvertUtils::CheckSubClass("infer type", input_args[0]->BuildType(), valid_types, p… in InferType()
|
D | exp.cc | 56 …(void)CheckAndConvertUtils::CheckSubClass("x_type", input_args[0]->BuildType(), valid_params_types… in InferType()
|
D | reciprocal.cc | 57 …(void)CheckAndConvertUtils::CheckSubClass("x_type", input_args[0]->BuildType(), valid_params_types… in InferType()
|
D | neg.cc | 57 …(void)CheckAndConvertUtils::CheckSubClass("x_type", input_args[0]->BuildType(), valid_params_types… in InferType()
|
D | log.cc | 57 …(void)CheckAndConvertUtils::CheckSubClass("x_type", input_args[0]->BuildType(), valid_params_types… in InferType()
|
D | unpack.cc | 30 …(void)CheckAndConvertUtils::CheckSubClass("x", input_args[0]->BuildType(), {TypeIdToType(kObjectTy… in UnpackInfer()
|
D | zeros.cc | 50 return CheckAndConvertUtils::CheckSubClass("dtype", output_type, valid_types, prim_name); in ZerosInferType()
|
D | assert.cc | 74 (void)CheckAndConvertUtils::CheckSubClass("input", dtype, template_types, op_name); in AssertInfer()
|
D | ones.cc | 48 return CheckAndConvertUtils::CheckSubClass("dtype", output_type, valid_types, prim_name); in OnesInferType()
|
D | masked_fill.cc | 65 …(void)CheckAndConvertUtils::CheckSubClass("value", input_args[kInputIndex2]->BuildType(), {kFloat}… in InferType()
|
D | pad.cc | 52 …return CheckAndConvertUtils::CheckSubClass("infer type", input_args[0]->BuildType(), valid_types, … in InferType()
|
D | lerp.cc | 64 …(void)CheckAndConvertUtils::CheckSubClass("weight", input_args[kInputIndex2]->BuildType(), {kFloat… in InferType()
|
D | gather_d.cc | 70 …(void)CheckAndConvertUtils::CheckSubClass("dim", input_args[kInputIndex1]->BuildType(), valid_type… in GatherDInfer()
|
D | pack.cc | 25 …(void)CheckAndConvertUtils::CheckSubClass("input_x[0]", x_types[0], {TypeIdToType(kObjectTypeTenso… in _get_pack_shape()
|
D | broadcast_to.cc | 71 (void)CheckAndConvertUtils::CheckSubClass("x_dtype", x_dtype, template_types, prim->name()); in BroadcastToInferType()
|
D | transpose.cc | 86 …return CheckAndConvertUtils::CheckSubClass("x", input_args[0]->BuildType(), {kTensorType}, prim->n… in InferType()
|
D | gather.cc | 40 …(void)CheckAndConvertUtils::CheckSubClass("params_type", input_args[kInputIndex0]->BuildType(), va… in GatherInfer()
|
/third_party/mindspore/mindspore/core/utils/ |
D | check_convert_utils.cc | 491 return CheckSubClass(input_names, check_type, check_list, prim_name); in CheckTensorTypeSame() 512 return CheckSubClass(type_name, type, check_list, prim_name); in CheckTensorTypeValid() 548 TypePtr CheckAndConvertUtils::CheckSubClass(const std::string &type_name, const TypePtr &type, in CheckSubClass() function in mindspore::CheckAndConvertUtils 638 return CheckSubClass(arg_name, arg_type, valid_type, prim_name); in CheckTypeValid()
|
D | check_convert_utils.h | 288 static TypePtr CheckSubClass(const std::string &type_name, const TypePtr &type,
|