| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | bias_add_info.cc | 83 Shapes tmp_inputs_shape = {inputs_shape_[0], inputs_shape_[0]}; in GenerateOpStrategies() local
85 …if (GenerateStrategiesForIndependentInputs(stage_id, tmp_inputs_shape, tmp_splittable_inputs, &sp_… in GenerateOpStrategies()
|
| D | select_info.cc | 106 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
109 …if (GenerateStrategiesForIndependentInputs(stage_id, tmp_inputs_shape, splittable_input, &sp_vecto… in GenerateOpStrategies()
|
| D | gathernd_info.cc | 130 Shapes tmp_inputs_shape = {inputs_shape_[1]}; in GenerateOpStrategies() local
133 …if (GenerateStrategiesForIndependentInputs(stage_id, tmp_inputs_shape, splittable_input, &sp_vecto… in GenerateOpStrategies()
|
| D | pack_info.cc | 141 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
144 …if (GenerateStrategiesForIndependentInputs(stage_id, tmp_inputs_shape, splittable_input, &sp_vecto… in GenerateOpStrategies()
|
| D | topk_info.cc | 158 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
161 …if (GenerateStrategiesForIndependentInputs(stage_id, tmp_inputs_shape, splittable_input, &sp_vecto… in GenerateOpStrategies()
|
| D | scatter_update_info.cc | 149 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
152 …if (GenerateStrategiesForIndependentInputs(stage_id, tmp_inputs_shape, splittable_input, &sp_vecto… in GenerateOpStrategies()
|
| D | concat_info.cc | 163 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
166 …if (GenerateStrategiesForIndependentInputs(stage_id, tmp_inputs_shape, splittable_input, &sp_vecto… in GenerateOpStrategies()
|
| D | split_info.cc | 149 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
152 …if (GenerateStrategiesForIndependentInputs(stage_id, tmp_inputs_shape, splittable_input, &sp_vecto… in GenerateOpStrategies()
|
| D | dsd_matmul_info.cc | 183 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
186 …if (GenerateStrategiesForIndependentInputs(stage_id, tmp_inputs_shape, splittable_input, &sp_vecto… in GenerateOpStrategies()
|
| D | broadcast_to_info.cc | 132 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
135 …if (GenerateStrategiesForIndependentInputs(stage_id, tmp_inputs_shape, splittable_input, &sp_vecto… in GenerateOpStrategies()
|
| D | tile_info.cc | 204 Shapes tmp_inputs_shape = {full_multiples_}; in GenerateOpStrategies() local
205 …if (GenerateStrategiesForIndependentInputs(stage_id, tmp_inputs_shape, splittable_inputs, &sp_vect… in GenerateOpStrategies()
|
| D | gatherd_info.cc | 198 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
201 …if (GenerateStrategiesForIndependentInputs(stage_id, tmp_inputs_shape, splittable_inputs, &sp_vect… in GenerateOpStrategies()
|
| D | layer_norm_info.cc | 218 Shapes tmp_inputs_shape = {input_shape_}; in GenerateOpStrategies() local
220 …if (GenerateStrategiesForIndependentInputs(stage_id, tmp_inputs_shape, splittable_inputs, &sp_vect… in GenerateOpStrategies()
|
| D | matmul_dds_info.cc | 291 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
294 …if (GenerateStrategiesForIndependentInputs(stage_id, tmp_inputs_shape, splittable_input, &sp_vecto… in GenerateOpStrategies()
|
| D | operator_info.cc | 1107 Shapes tmp_inputs_shape = {inputs_shape[1], inputs_shape[1]}; in GenerateStrategiesForBroadcastLeft() local
1109 …if (GenerateStrategiesForTwoEqualInputs(stage_id, tmp_inputs_shape, tmp_splittable_inputs, sp_vect… in GenerateStrategiesForBroadcastLeft()
1156 Shapes tmp_inputs_shape = {inputs_shape[0], inputs_shape[0]}; in GenerateStrategiesForBroadcastRight() local
1158 …if (GenerateStrategiesForTwoEqualInputs(stage_id, tmp_inputs_shape, tmp_splittable_inputs, sp_vect… in GenerateStrategiesForBroadcastRight()
1218 Shapes tmp_inputs_shape = {max_shape, max_shape}; in GenerateStrategiesForBroadcastBoth() local
1220 …if (GenerateStrategiesForTwoEqualInputs(stage_id, tmp_inputs_shape, tmp_splittable_inputs, sp_vect… in GenerateStrategiesForBroadcastBoth()
|