Searched refs:tmp_splittable_inputs (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | bias_add_info.cc | 84 Shapes tmp_splittable_inputs = {splittable_inputs[0], splittable_inputs[0]}; in GenerateOpStrategies() local
85 …if (GenerateStrategiesForIndependentInputs(stage_id, tmp_inputs_shape, tmp_splittable_inputs, &sp_… in GenerateOpStrategies()
|
| D | operator_info.cc | 1108 Shapes tmp_splittable_inputs = {splittable_inputs[1], splittable_inputs[1]}; in GenerateStrategiesForBroadcastLeft() local
1109 …if (GenerateStrategiesForTwoEqualInputs(stage_id, tmp_inputs_shape, tmp_splittable_inputs, sp_vect… in GenerateStrategiesForBroadcastLeft()
1157 Shapes tmp_splittable_inputs = {splittable_inputs[0], splittable_inputs[0]}; in GenerateStrategiesForBroadcastRight() local
1158 …if (GenerateStrategiesForTwoEqualInputs(stage_id, tmp_inputs_shape, tmp_splittable_inputs, sp_vect… in GenerateStrategiesForBroadcastRight()
1219 Shapes tmp_splittable_inputs = {splittable_vector, splittable_vector}; in GenerateStrategiesForBroadcastBoth() local
1220 …if (GenerateStrategiesForTwoEqualInputs(stage_id, tmp_inputs_shape, tmp_splittable_inputs, sp_vect… in GenerateStrategiesForBroadcastBoth()
|