| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | uniform_candidate_sampler_info.cc | 93 Dimensions input_strategy = stra.at(0); in CheckStrategy() local
180 Dimensions input_strategy(inputs_shape_[0].size(), 1); in GenerateBatchStrategies() local
187 auto input_strategy = input_dim.at(0); in replace_graph() local
200 auto input_strategy = input_dim.at(0); in ComputeReplaceGraph() local
|
| D | loss_info.cc | 37 Dimensions input_strategy = stra.at(0); in CheckStrategy() local
73 Dimensions input_strategy = stra.at(0); in InferDevMatrixShape() local
|
| D | elementary_function_info.cc | 37 Dimensions input_strategy = stra.at(0); in CheckStrategy() local
|
| D | group_norm_info.cc | 60 Dimensions input_strategy = strategies[INPUT_INDEX]; in CheckStrategy() local
160 Shape input_strategy(this->inputs_shape_[INPUT_INDEX].size(), 1); in GenerateOpStrategies() local
|
| D | l2_normalize_info.cc | 36 Dimensions input_strategy = stra.at(0); in CheckStrategy() local
|
| D | tmp_identity_info.cc | 34 Dimensions input_strategy = stra.at(0); in InferDevMatrixShape() local
|
| D | prelu_info.cc | 58 Dimensions input_strategy = stra.at(0); in InferDevMatrixShape() local
|
| D | reduce_method_info.cc | 38 Dimensions input_strategy = stra.at(0); in InferDevMatrixShape() local
343 Shape input_strategy = stra.at(0); in CheckStrategy() local
414 Dimensions input_strategy(inputs_shape_[0].size(), 1); in GenerateBatchStrategies() local
480 Shape input_strategy = stra.at(0); in CheckStrategy() local
|
| D | random_choice_with_mask_info.cc | 41 Dimensions input_strategy = strategies[0]; in CheckStrategy() local
|
| D | maxpool_info.cc | 148 Dimensions input_strategy = stra[0]; in CheckStrategy() local
|
| D | gathernd_info.cc | 145 Dimensions input_strategy(inputs_shape_[0].size(), 1); in GenerateOpStrategies() local
|
| D | rms_norm_info.cc | 50 Dimensions input_strategy = stra[RMS_NORM_INPUT_INDEX]; in CheckStrategy() local
186 Dimensions input_strategy = sp->GetInputDim()[0]; in GenerateGammaStrategies() local
|
| D | avgpool_info.cc | 162 Dimensions input_strategy = stra[0]; in CheckStrategy() local
|
| D | resizebilinear_v2_info.cc | 79 Dimensions input_strategy = stra[0]; in CheckStrategy() local
499 Dimensions input_strategy = stra[0]; in CheckStrategy() local
|
| D | pad_info.cc | 92 auto input_strategy = stra[0]; in CheckStrategy() local
|
| D | dsd_matmul_info.cc | 95 Dimensions input_strategy = stra.at(0); in InferDevMatrixShape() local
|
| D | layer_norm_info.cc | 98 Dimensions input_strategy = stra[LAYER_NORM_INPUT_INDEX]; in CheckStrategy() local
263 Dimensions input_strategy = sp->GetInputDim()[0]; in GenerateGammaAndBetaStrategies() local
|
| D | topk_info.cc | 246 Dimensions input_strategy(inputs_shape_[0].size(), 1); in GenerateBatchStrategies() local
|
| D | split_info.cc | 228 Dimensions input_strategy(inputs_shape_[0].size(), 1); in GenerateBatchStrategies() local
|
| D | conv3d_info.cc | 85 Dimensions input_strategy = stra[0]; in CheckStrategy() local
|
| D | conv2d_info.cc | 374 Dimensions input_strategy = stra[0]; in CheckStrategy() local
1036 Shape input_strategy(inputs_shape_[0].size(), 1); in GenerateOpStrategies() local
1148 Dimensions input_strategy = stra[0]; in CheckStrategy() local
1586 Dimensions input_strategy = stra[0]; in CheckStrategy() local
|
| D | onehot_info.cc | 71 Dimensions input_strategy = stra.at(0); in InferDevMatrixShape() local
|
| D | activation_info.cc | 209 Dimensions input_strategy = stra.at(0); in CheckStrategy() local
416 Dimensions input_strategy = stra.at(0); in CheckStrategy() local
465 Dimensions input_strategy = stra.at(0); in InferDevMatrixShape() local
|
| D | get_next_info.cc | 123 Dimensions input_strategy; in CheckStrategy() local
|
| D | matmul_dds_info.cc | 134 Dimensions input_strategy = stra.at(0); in InferDevMatrixShape() local
|