| /third_party/mindspore/mindspore/core/ops/ |
| D | apply_adagrad_d_a.cc | 38 auto var_shape = input_args[0]->BuildShape(); in InferShape()
39 auto gradient_accumulator_shape = input_args[kInputIndex1]->BuildShape(); in InferShape()
40 auto gradient_squared_accumulator_shape = input_args[kInputIndex2]->BuildShape(); in InferShape()
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 | sparse_apply_r_m_s_prop.cc | 37 auto var_shape_ptr = input_args[0]->BuildShape(); in InferShape()
38 auto ms_shape_ptr = input_args[1]->BuildShape(); in InferShape()
39 auto mom_shape_ptr = input_args[2]->BuildShape(); in InferShape()
43 …auto lr_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[3]->BuildShape())[kShap… in InferShape()
44 …auto grad_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[4]->BuildShape())[kSh… in InferShape()
45 …auto indices_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[5]->BuildShape())[… in InferShape()
|
| D | assign.cc | 47 …e_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(args_spec_list[0]->BuildShape())[kShape]; in InferImplAssign()
48 …auto value_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(args_spec_list[1]->BuildShape()… in InferImplAssign()
62 << args_spec_list[1]->BuildShape()->ToString() in InferImplAssign()
64 << args_spec_list[0]->BuildShape()->ToString() << "."; in InferImplAssign()
|
| D | select.cc | 46 …cond_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kCondIndex]->BuildShape()); in SelectInferShape()
47 auto x_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kXIndex]->BuildShape()); in SelectInferShape()
48 auto y_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kYIndex]->BuildShape()); in SelectInferShape()
62 MS_LOG(ERROR) << " cond shape :" << input_args[kCondIndex]->BuildShape()->ToString(); in SelectInferShape()
63 MS_LOG(ERROR) << " x shape :" << input_args[kXIndex]->BuildShape()->ToString(); in SelectInferShape()
64 MS_LOG(ERROR) << " y shape :" << input_args[kYIndex]->BuildShape()->ToString(); in SelectInferShape()
67 return input_args[1]->BuildShape(); in SelectInferShape()
|
| 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()
72 auto input_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in InferShape()
|
| D | tensor_list_stack.cc | 55 …auto input0_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[k… in TensorListStackInfer()
60 if (input_args[1]->BuildShape() == nullptr) { in TensorListStackInfer()
63 …auto input1_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->BuildShape())[k… in TensorListStackInfer()
|
| D | batch_norm.cc | 79 …auto input_x = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kShape… in BatchNormInfer()
84 …auto scale = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->BuildShape(… in BatchNormInfer()
85 …auto bias = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildShape()… in BatchNormInfer()
86 …auto mean = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex3]->BuildShape()… in BatchNormInfer()
87 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex4]->BuildShape())[kShape]; in BatchNormInfer()
|
| 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 | masked_fill.cc | 32 …hape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->BuildShape()); in InferShape()
34 …hape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->BuildShape()); in InferShape()
36 …hape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildShape()); 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 | lerp.cc | 36 …hape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->BuildShape()); in InferShape()
38 …hape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->BuildShape()); in InferShape()
40 …hape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildShape()); 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 …auto idx_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->BuildShape())[kSha… in IndexAddInferShape()
|
| D | transpose.cc | 31 …auto x_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kShape… in InferShape()
32 …auto x_min_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kM… in InferShape()
33 …auto x_max_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kM… in InferShape()
|
| D | sigmoid_cross_entropy_with_logits.cc | 38 …auto x_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kShape… in SigmoidCrossEntropyWithLogitsInfer()
39 …auto y_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->BuildShape())[kShape… in SigmoidCrossEntropyWithLogitsInfer()
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | cdist_grad.cc | 32 …auto grad_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kSh… in InferShape()
33 …auto x_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->BuildShape())[kShape… in InferShape()
34 …auto y_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[2]->BuildShape())[kShape… in InferShape()
35 …auto cdist_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[3]->BuildShape())[kS… 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 | layer_norm_grad.cc | 35 x_backprob->BuildShape(), gamma_backprob->BuildShape(), beta_backprob->BuildShape()}); in LayerNormGradInfer()
|
| 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 …auto dloss = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildShape(… in SmoothL1LossGradInfer()
|
| D | soft_margin_loss_grad.cc | 32 … = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->BuildShape())[kShape]; in SoftMarginLossGradInferShape()
33 …auto label = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->BuildShape(… in SoftMarginLossGradInferShape()
34 …auto dout = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildShape()… 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/tests/ut/cpp/pipeline/static_analysis/ |
| D | data_test.cc | 105 ASSERT_TRUE(NoShape() == *s1->BuildShape()); in TEST_F()
108 ASSERT_TRUE(NoShape() == *f1->BuildShape()); in TEST_F()
111 auto lshape = l1.BuildShape(); in TEST_F()
121 ShapePtr shape_weight = dyn_cast<Shape>(abstract_weight1->BuildShape()); in TEST_F()
127 std::shared_ptr<TupleShape> shape_tuple = dyn_cast<TupleShape>(abstract_tup->BuildShape()); in TEST_F()
|
| /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()
|