| /third_party/mindspore/mindspore/core/ops/ |
| D | apply_adagrad_d_a.cc | 67 auto var_type = input_args[kInputIndex0]->BuildType(); in InferType()
68 auto gradient_accumulator_type = input_args[kInputIndex1]->BuildType(); in InferType()
69 auto gradient_squared_accumulator_type = input_args[kInputIndex2]->BuildType(); in InferType()
70 auto grad_type = input_args[kInputIndex3]->BuildType(); in InferType()
71 auto lr_type = input_args[kInputIndex4]->BuildType(); in InferType()
72 auto l1_type = input_args[kInputIndex5]->BuildType(); in InferType()
73 auto l2_type = input_args[kInputIndex6]->BuildType(); in InferType()
74 auto global_step_type = input_args[kInputIndex7]->BuildType(); in InferType()
|
| D | masked_fill.cc | 58 …(void)CheckAndConvertUtils::CheckTensorTypeValid("mask", input_args[1]->BuildType(), {kBool}, op_n… in InferType()
61 (void)types.emplace("input", input_args[kInputIndex0]->BuildType()); in InferType()
62 (void)types.emplace("value", input_args[kInputIndex2]->BuildType()); in InferType()
65 …(void)CheckAndConvertUtils::CheckSubClass("value", input_args[kInputIndex2]->BuildType(), {kFloat}… in InferType()
66 return CheckAndConvertUtils::CheckTensorTypeValid("input", input_args[0]->BuildType(), in InferType()
|
| D | apply_momentum.cc | 74 auto v_tensor_type = input_args[kInputIndex0]->BuildType(); in ApplyMomentumInfer()
75 auto a_tensor_type = input_args[kInputIndex1]->BuildType(); in ApplyMomentumInfer()
76 auto l_type = input_args[kInputIndex2]->BuildType(); in ApplyMomentumInfer()
77 auto g_type = input_args[kInputIndex3]->BuildType(); in ApplyMomentumInfer()
78 auto m_type = input_args[kInputIndex4]->BuildType(); in ApplyMomentumInfer()
|
| D | batch_norm.cc | 110 auto scale_type = input_args[kInputIndex1]->BuildType()->cast<TensorTypePtr>()->element(); in BatchNormInfer()
111 auto bias_type = input_args[kInputIndex2]->BuildType()->cast<TensorTypePtr>()->element(); in BatchNormInfer()
115 …CheckAndConvertUtils::CheckTensorTypeValid("x", input_args[kInputIndex0]->BuildType(), valid_types… in BatchNormInfer()
117 (void)args.emplace("scale", input_args[kInputIndex1]->BuildType()); in BatchNormInfer()
118 (void)args.emplace("bias", input_args[kInputIndex2]->BuildType()); in BatchNormInfer()
121 (void)args_moving.emplace("scale", input_args[kInputIndex2]->BuildType()); in BatchNormInfer()
122 (void)args_moving.emplace("bias", input_args[kInputIndex3]->BuildType()); in BatchNormInfer()
|
| D | sparse_apply_r_m_s_prop.cc | 79 auto var_type = input_args[0]->BuildType(); in InferType()
80 auto ms_type = input_args[1]->BuildType(); in InferType()
81 auto mom_type = input_args[2]->BuildType(); in InferType()
82 auto lr_type = input_args[3]->BuildType(); in InferType()
83 auto grad_type = input_args[4]->BuildType(); in InferType()
84 auto indices_type = input_args[5]->BuildType(); in InferType()
|
| D | binary_cross_entropy.cc | 61 (void)types.emplace("x_shape", input_args[kInputIndex0]->BuildType()); in BinaryCrossEntroyInferType()
62 (void)types.emplace("y_shape", input_args[kInputIndex1]->BuildType()); in BinaryCrossEntroyInferType()
64 if (input_args[kInputIndex3]->BuildType() != nullptr) { in BinaryCrossEntroyInferType()
65 (void)types.emplace("x_shape", input_args[kInputIndex0]->BuildType()); in BinaryCrossEntroyInferType()
66 (void)types.emplace("weight_shape", input_args[kInputIndex2]->BuildType()); in BinaryCrossEntroyInferType()
|
| D | scatter_nd_update.cc | 42 auto input_x_type = input_args[kInputIndex0]->BuildType(); in InferType()
43 auto indices_type = input_args[kInputIndex1]->BuildType(); in InferType()
44 auto update_type = input_args[kInputIndex2]->BuildType(); in InferType()
48 return input_args[0]->BuildType(); in InferType()
|
| D | one_hot.cc | 43 …(void)CheckAndConvertUtils::CheckTensorTypeValid("depth", input_args[1]->BuildType(), {kInt64}, op… in OneHotInferShape()
77 …(void)CheckAndConvertUtils::CheckTensorTypeValid("indices", input_args[kInputIndex0]->BuildType(),… in OneHotInferType()
79 (void)CheckAndConvertUtils::CheckTypeValid("depth", input_args[kInputIndex1]->BuildType(), in OneHotInferType()
81 std::map<std::string, TypePtr> args = {{"on_value", input_args[kInputIndex2]->BuildType()}, in OneHotInferType()
82 {"off_dtype", input_args[kInputIndex3]->BuildType()}}; in OneHotInferType()
|
| D | layer_norm.cc | 72 …(void)CheckAndConvertUtils::CheckTensorTypeValid("x_dtype", input_args[x_index]->BuildType(), vali… in LayerNormInfer()
73 …onvertUtils::CheckTensorTypeValid("gamma_dtype", input_args[gamma_index]->BuildType(), valid_types, in LayerNormInfer()
75 …(void)CheckAndConvertUtils::CheckTensorTypeValid("beta_dtype", input_args[beta_index]->BuildType()… in LayerNormInfer()
105 …std::vector<TypePtr> types_list = {input_x->BuildType(), input_x->BuildType(), input_x->BuildType(… in LayerNormInfer()
|
| D | addn.cc | 77 (void)types.emplace("element_0", elements[0]->BuildType()); in AddNInferType()
79 if (elements[i]->BuildType()->type_id() == kObjectTypeUndeterminedType) { in AddNInferType()
80 return elements[0]->BuildType(); in AddNInferType()
83 (void)types.emplace(element_i, elements[i]->BuildType()); in AddNInferType()
88 return elements[0]->BuildType(); in AddNInferType()
|
| D | stack.cc | 49 auto infer_type0 = input_args[0]->BuildType()->cast<TensorTypePtr>()->element(); in StackInfer()
51 if (input_args[i]->BuildType()->cast<TensorTypePtr>()->element() == infer_type0) { in StackInfer()
53 … << "] data type = " << input_args[i]->BuildType()->cast<TensorTypePtr>()->element(); in StackInfer()
56 auto infer_type = input_args[0]->BuildType()->cast<TensorTypePtr>()->element(); in StackInfer()
|
| D | lerp.cc | 59 (void)types.emplace("start", input_args[0]->BuildType()); in InferType()
60 (void)types.emplace("end", input_args[1]->BuildType()); in InferType()
62 (void)types.emplace("weight", input_args[kInputIndex2]->BuildType()); in InferType()
64 …(void)CheckAndConvertUtils::CheckSubClass("weight", input_args[kInputIndex2]->BuildType(), {kFloat… in InferType()
|
| D | gather.cc | 40 …(void)CheckAndConvertUtils::CheckSubClass("params_type", input_args[kInputIndex0]->BuildType(), va… in GatherInfer()
43 …dConvertUtils::CheckTensorTypeValid("index_type", input_args[kInputIndex1]->BuildType(), int_types, in GatherInfer()
45 …(void)CheckAndConvertUtils::CheckTypeValid("axis_type", input_args[kInputIndex2]->BuildType(), int… in GatherInfer()
94 params->BuildType()); in GatherInfer()
96 return abstract::MakeAbstract(std::make_shared<abstract::Shape>(out_shape), params->BuildType()); in GatherInfer()
|
| D | adam.cc | 40 auto var_type = input_args[kInputIndex0]->BuildType(); in AdamInfer()
41 auto m_type = input_args[kInputIndex1]->BuildType(); in AdamInfer()
42 auto v_type = input_args[kInputIndex2]->BuildType(); in AdamInfer()
43 auto grad_type = input_args[kInputIndex9]->BuildType(); in AdamInfer()
|
| D | scatter_nd.cc | 45 auto indices_type = input_args[0]->BuildType(); in InferType()
46 auto update_type = input_args[1]->BuildType(); in InferType()
49 return input_args[1]->BuildType(); in InferType()
|
| D | soft_margin_loss.cc | 49 (void)types.emplace("logits", input_args[0]->BuildType()); in SoftMarginLossInferType()
50 (void)types.emplace("labels", input_args[1]->BuildType()); in SoftMarginLossInferType()
52 return input_args[0]->BuildType(); in SoftMarginLossInferType()
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | binary_cross_entropy_grad.cc | 43 (void)types.emplace("x_shape", input_args[kInputIndex0]->BuildType()); in BinaryCrossEntroyGradInferType()
44 (void)types.emplace("y_shape", input_args[kInputIndex1]->BuildType()); in BinaryCrossEntroyGradInferType()
46 if (input_args[kInputIndex3]->BuildType() != nullptr) { in BinaryCrossEntroyGradInferType()
47 (void)types.emplace("x_shape", input_args[kInputIndex0]->BuildType()); in BinaryCrossEntroyGradInferType()
48 (void)types.emplace("weight_shape", input_args[kInputIndex2]->BuildType()); in BinaryCrossEntroyGradInferType()
|
| D | soft_margin_loss_grad.cc | 47 (void)types.emplace("logits", input_args[kInputIndex0]->BuildType()); in SoftMarginLossGradInferType()
48 (void)types.emplace("labels", input_args[kInputIndex1]->BuildType()); in SoftMarginLossGradInferType()
49 (void)types.emplace("dout", input_args[kInputIndex2]->BuildType()); in SoftMarginLossGradInferType()
51 …return CheckAndConvertUtils::CheckTensorTypeValid("logits", input_args[0]->BuildType(), valid_type… in SoftMarginLossGradInferType()
|
| D | cdist_grad.cc | 53 (void)types.emplace("grad", input_args[0]->BuildType()); in InferType()
54 (void)types.emplace("input_x", input_args[1]->BuildType()); in InferType()
55 (void)types.emplace("input_y", input_args[2]->BuildType()); in InferType()
56 (void)types.emplace("cdist", input_args[3]->BuildType()); in InferType()
|
| D | sigmoid_cross_entropy_with_logits_grad.cc | 49 (void)args.emplace("x_type", input_args[kInputIndex0]->BuildType()); in SigmoidCrossEntropyWithLogitsGradInfer()
50 (void)args.emplace("y_type", input_args[kInputIndex1]->BuildType()); in SigmoidCrossEntropyWithLogitsGradInfer()
51 (void)args.emplace("dout_type", input_args[kInputIndex2]->BuildType()); in SigmoidCrossEntropyWithLogitsGradInfer()
|
| D | layer_norm_grad.cc | 37 …std::vector<TypePtr>{x_backprob->BuildType(), gamma_backprob->BuildType(), beta_backprob->BuildTyp… in LayerNormGradInfer()
|
| D | smooth_l1_loss_grad.cc | 56 (void)args.emplace("prediction", input_args[kInputIndex0]->BuildType()); in SmoothL1LossGradInfer()
57 (void)args.emplace("target", input_args[kInputIndex1]->BuildType()); in SmoothL1LossGradInfer()
58 (void)args.emplace("dloss", input_args[kInputIndex2]->BuildType()); in SmoothL1LossGradInfer()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | abstract_value.h | 73 virtual TypePtr BuildType() const = 0;
121 TypePtr BuildType() const override { return GetTypeTrack(); } in BuildType() function
143 TypePtr BuildType() const override { return std::make_shared<TypeType>(); } in BuildType() function
159 TypePtr BuildType() const override { return std::make_shared<Problem>(); } in MS_DECLARE_PARENT()
184 TypePtr BuildType() const override { return GetTypeTrack(); } in BuildType() function
213 TypePtr BuildType() const override { return std::make_shared<Function>(); } in BuildType() function
243 TypePtr BuildType() const override;
303 TypePtr BuildType() const override { return std::make_shared<UndeterminedType>(); } in MS_DECLARE_PARENT()
332 TypePtr BuildType() const override;
400 TypePtr BuildType() const override { return std::make_shared<Tuple>(ElementsType()); } in MS_DECLARE_PARENT()
[all …]
|
| D | param_validator.cc | 55 TypePtr type = tensor->BuildType(); in CheckTensorDType()
62 TypePtr ele_type = elem->BuildType(); in CheckTensorDType()
79 TypePtr sample_type = sample_elem->BuildType(); in CheckTensorsDTypeSame()
88 auto a_type = elem->BuildType(); in CheckTensorsDTypeSame()
105 auto type = scalar->BuildType(); in CheckScalarType()
131 TypePtr type_base = base_elem->BuildType(); in CheckDtypeSame()
135 TypePtr type = tensor_elem->BuildType(); in CheckDtypeSame()
|
| D | abstract_value.cc | 92 auto type = BuildType(); in operator ==()
93 auto other_type = BuildType(); in operator ==()
291 TypePtr element_type = ele->BuildType(); in ElementsType()
435 TypePtr AbstractSlice::BuildType() const { in BuildType() function in mindspore::abstract::AbstractSlice
439 TypePtr start = start_->BuildType(); in BuildType()
440 TypePtr stop = stop_->BuildType(); in BuildType()
441 TypePtr step = step_->BuildType(); in BuildType()
532 TypePtr AbstractTensor::BuildType() const { in BuildType() function in mindspore::abstract::AbstractTensor
534 TypePtr element_type = element_->BuildType(); in BuildType()
549 auto type = other->BuildType(); in Join()
[all …]
|