| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/parallel/ops_info/ |
| D | onehot_info_test_axis_0.cc | 139 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
161 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
183 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
|
| D | onehot_info_test.cc | 139 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
161 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
183 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
|
| D | reshape_test.cc | 106 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
127 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
|
| D | prelu_test.cc | 93 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
198 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
|
| D | tanh_info_test.cc | 90 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
|
| D | pow_info_test.cc | 90 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
|
| D | gelu_info_test.cc | 90 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
|
| D | log_softmax_info_test.cc | 91 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
|
| D | tmpidentity_test.cc | 92 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
|
| D | activation_info_test.cc | 91 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
|
| D | softmax_entropy_loss_info_test.cc | 95 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
|
| D | softmax_info_test.cc | 95 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
|
| D | l2_normalize_info_test.cc | 91 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
|
| D | transpose_test.cc | 106 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
|
| D | reduce_method_test.cc | 96 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | dropout_do_mask_info.cc | 169 void SetGenMaskShape(const CNodePtr &cnode, const Shape &input_slice_shape) { in SetGenMaskShape()
231 Shape input_slice_shape = inputs_tensor_info_[0].slice_shape(); in GetDropoutGenMaskReplaceOp() local
|
| D | slice_info.cc | 209 Shape input_slice_shape = inputs_tensor_info_[0].slice_shape(); in ComputeReplaceGraph() local
|
| D | scatter_nd_ops_info.cc | 282 auto input_slice_shape = inputs_tensor_info_[0].slice_shape(); in PrepareReplaceGraph() local
|
| D | flash_attention_score_info.cc | 145 …Shape input_slice_shape = inputs_tensor_info_.at(ops::kFlashAttentionScoreInputDropMaskIndex).slic… in UpdateDropoutGenMaskSliceShapeAndSeed() local
1051 …Shape input_slice_shape = inputs_tensor_info_.at(ops::kFlashAttentionScoreInputDropMaskIndex).slic… in ReplaceNodeInputOrAttrs() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/parallel/auto_parallel/ |
| D | operator_costmodel_test.cc | 171 Shape input_slice_shape = {8, 32, 8, 8}; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | operator_costmodel.cc | 1208 Shape input_slice_shape = input_tensor_info.slice_shape(); in GetBackwardCommCost() local
1537 Shape input_slice_shape = input.slice_shape(); in GetBackwardCommCost() local
1551 Shape input_slice_shape = inputs[0].slice_shape(); in GetForwardComputationCost() local
1566 Shape input_slice_shape = input.slice_shape(); in GetBackwardComputationCost() local
1995 Shape input_slice_shape = input_tensor_info.slice_shape(); in GetBackwardCommCost() local
|