| /third_party/mindspore/mindspore/core/ops/ |
| D | sparse_apply_r_m_s_prop.cc | 40 auto var_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(var_shape_ptr)[kShape]; in InferShape()
41 auto ms_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(ms_shape_ptr)[kShape]; in InferShape()
42 auto mom_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(mom_shape_ptr)[kShape]; in InferShape()
43 …to lr_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[3]->BuildShape())[kShape]; in InferShape()
44 … grad_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[4]->BuildShape())[kShape]; in InferShape()
45 …dices_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[5]->BuildShape())[kShape]; in InferShape()
|
| D | batch_norm.cc | 79 …uto input_x = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kShape]; in BatchNormInfer()
84 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->BuildShape())[kShape]; in BatchNormInfer()
85 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildShape())[kShape]; in BatchNormInfer()
86 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex3]->BuildShape())[kShape]; in BatchNormInfer()
87 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex4]->BuildShape())[kShape]; in BatchNormInfer()
91 …ape_norm = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->GetShapeTrack())[kShape]; in BatchNormInfer()
|
| D | ctcloss.cc | 41 auto inputs_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(inputs->BuildShape())[kShape]; in CheckCTCLossInputs()
42 …ices_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(labels_indices->BuildShape())[kShape]; in CheckCTCLossInputs()
43 …alues_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(labels_values->BuildShape())[kShape]; in CheckCTCLossInputs()
44 …gth_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(sequence_length->BuildShape())[kShape]; in CheckCTCLossInputs()
73 auto shape = input_shape[kShape]; in InferShape()
|
| D | adam.cc | 31 …CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->GetShapeTrack())[kShape]; in AdamInfer()
32 …CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->GetShapeTrack())[kShape]; in AdamInfer()
33 …CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->GetShapeTrack())[kShape]; in AdamInfer()
34 …CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex9]->GetShapeTrack())[kShape]; in AdamInfer()
|
| D | broadcast_to.cc | 27 …uto x_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kShape]; in BroadcastToInferShape()
28 auto value_ptr = primitive->GetAttr(kShape); in BroadcastToInferShape()
80 (void)AddAttr(kShape, MakeValue(shape)); in set_shape()
84 auto value_ptr = GetAttr(kShape); in get_shape()
|
| D | scatter_nd_update.cc | 27 …CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->GetShapeTrack())[kShape]; in InferShape()
28 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->BuildShape())[kShape]; in InferShape()
29 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildShape())[kShape]; in InferShape()
|
| D | lsh_projection.cc | 39 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->BuildShape())[kShape]; in LshProjectionInfer()
40 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->BuildShape())[kShape]; in LshProjectionInfer()
46 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildShape())[kShape]; in LshProjectionInfer()
|
| D | where.cc | 38 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->BuildShape())[kShape]; in WhereInfer()
40 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->BuildShape())[kShape]; in WhereInfer()
42 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildShape())[kShape]; in WhereInfer()
|
| D | lerp.cc | 37 auto start_shape = start_shape_map[kShape]; in InferShape()
39 auto end_shape = end_shape_map[kShape]; in InferShape()
41 auto weight_shape = weight_shape_map[kShape]; in InferShape()
|
| D | fake_quant_with_min_max_vars_per_channel.cc | 47 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->BuildShape())[kShape]; in FakeQuantWithMinMaxVarsPerChannelInfer()
48 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->BuildShape())[kShape]; in FakeQuantWithMinMaxVarsPerChannelInfer()
49 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildShape())[kShape]; in FakeQuantWithMinMaxVarsPerChannelInfer()
|
| D | masked_fill.cc | 33 auto input_shape = input_shape_map[kShape]; in InferShape()
35 auto mask_shape = mask_shape_map[kShape]; in InferShape()
37 auto value_shape = value_shape_map[kShape]; in InferShape()
|
| D | apply_adagrad_d_a.cc | 41 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex4]->BuildShape())[kShape]; in InferShape()
42 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex5]->BuildShape())[kShape]; in InferShape()
43 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex6]->BuildShape())[kShape]; in InferShape()
45 CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex7]->BuildShape())[kShape]; in InferShape()
|
| D | fake_quant_with_min_max_vars.cc | 32 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->BuildShape())[kShape]; in InferShape()
33 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->BuildShape())[kShape]; in InferShape()
34 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildShape())[kShape]; in InferShape()
|
| D | binary_cross_entropy.cc | 37 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->BuildShape())[kShape]; in BinaryCrossEntroyInferShape()
38 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->BuildShape())[kShape]; in BinaryCrossEntroyInferShape()
39 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildShape())[kShape]; in BinaryCrossEntroyInferShape()
|
| D | index_add.cc | 34 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->BuildShape())[kShape]; in IndexAddInferShape()
35 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildShape())[kShape]; in IndexAddInferShape()
41 …o idx_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->BuildShape())[kShape]; in IndexAddInferShape()
|
| D | sigmoid_cross_entropy_with_logits.cc | 38 …uto x_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kShape]; in SigmoidCrossEntropyWithLogitsInfer()
39 …uto y_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->BuildShape())[kShape]; in SigmoidCrossEntropyWithLogitsInfer()
|
| D | smooth_l1_loss.cc | 44 … prediction = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kShape]; in SmoothL1LossInfer()
45 …auto target = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kShape]; in SmoothL1LossInfer()
|
| D | softmax_cross_entropy_with_logits.cc | 36 …ogits_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kShape]; in SoftmaxCrossEntropyWithLogitsInfer()
37 …abels_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->BuildShape())[kShape]; in SoftmaxCrossEntropyWithLogitsInfer()
|
| D | select.cc | 50 if (x_shape[kShape] != cond_shape[kShape] || x_shape[kShape] != y_shape[kShape]) { in SelectInferShape()
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | cdist_grad.cc | 32 … grad_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kShape]; in InferShape()
33 …uto x_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->BuildShape())[kShape]; in InferShape()
34 …uto y_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[2]->BuildShape())[kShape]; in InferShape()
35 …cdist_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[3]->BuildShape())[kShape]; in InferShape()
|
| D | sigmoid_cross_entropy_with_logits_grad.cc | 39 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->BuildShape())[kShape]; in SigmoidCrossEntropyWithLogitsGradInfer()
40 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->BuildShape())[kShape]; in SigmoidCrossEntropyWithLogitsGradInfer()
41 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildShape())[kShape]; in SigmoidCrossEntropyWithLogitsGradInfer()
|
| D | smooth_l1_loss_grad.cc | 46 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->BuildShape())[kShape]; in SmoothL1LossGradInfer()
47 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->BuildShape())[kShape]; in SmoothL1LossGradInfer()
48 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildShape())[kShape]; in SmoothL1LossGradInfer()
|
| D | soft_margin_loss_grad.cc | 32 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->BuildShape())[kShape]; in SoftMarginLossGradInferShape()
33 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->BuildShape())[kShape]; in SoftMarginLossGradInferShape()
34 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildShape())[kShape]; in SoftMarginLossGradInferShape()
|
| D | binary_cross_entropy_grad.cc | 30 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->BuildShape())[kShape]; in BinaryCrossEntroyGradInferShape()
31 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->BuildShape())[kShape]; in BinaryCrossEntroyGradInferShape()
32 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildShape())[kShape]; in BinaryCrossEntroyGradInferShape()
|
| /third_party/mindspore/mindspore/core/ops/fusion/ |
| D | full_connection.cc | 69 auto input0_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input0->BuildShape())[kShape]; in FullConnectionInfer()
70 auto input1_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input1->BuildShape())[kShape]; in FullConnectionInfer()
98 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildShape())[kShape]; in FullConnectionInfer()
|