Home
last modified time | relevance | path

Searched refs:ResetInputs (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstrategy.h62 void ResetInputs(const Strategys &input) { inputs_ = input; } in ResetInputs() function
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dbias_add_info.cc100 sp->ResetInputs(tmp_strategy); in GenerateOpStrategies()
Dselect_info.cc123 sp->ResetInputs(tmp_strategy); in GenerateOpStrategies()
Dgathernd_info.cc147 sp->ResetInputs(tmp_strategy); in GenerateOpStrategies()
Dpack_info.cc158 sp->ResetInputs(tmp_strategy); in GenerateOpStrategies()
Dscatter_update_info.cc174 sp->ResetInputs(tmp_strategy); in GenerateOpStrategies()
Dconcat_info.cc180 sp->ResetInputs(tmp_strategy); in GenerateOpStrategies()
Ddsd_matmul_info.cc203 sp->ResetInputs(tmp_strategy); in GenerateOpStrategies()
Dbroadcast_to_info.cc149 sp->ResetInputs(tmp_strategy); in GenerateOpStrategies()
Dgatherd_info.cc214 sp->ResetInputs(tmp_strategy); in GenerateOpStrategies()
Dlayer_norm_info.cc197 sp->ResetInputs(tmp_strategy); in GenerateGammaAndBetaStrategies()
Dmatmul_dds_info.cc313 sp->ResetInputs(tmp_strategy); in GenerateOpStrategies()
Dunsorted_segment_op_info.cc203 sp->ResetInputs(tmp_strategy); in GenerateOpStrategies()
Doperator_info.cc1136 sp->ResetInputs(tmp_strategy); in GenerateStrategiesForBroadcastLeft()
1186 sp->ResetInputs(tmp_strategy); in GenerateStrategiesForBroadcastRight()
1238 sp->ResetInputs({input0_strategy, input1_strategy}); in GenerateStrategiesForBroadcastBoth()