Home
last modified time | relevance | path

Searched defs:tmp_inputs_shape (Results 1 – 15 of 15) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dselect_info.cc106 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Dgathernd_info.cc130 Shapes tmp_inputs_shape = {inputs_shape_[1]}; in GenerateOpStrategies() local
Dtopk_info.cc158 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Dscatter_update_info.cc149 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Dsplit_info.cc149 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Dbias_add_info.cc83 Shapes tmp_inputs_shape = {inputs_shape_[0], inputs_shape_[0]}; in GenerateOpStrategies() local
Dpack_info.cc141 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Dconcat_info.cc163 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Ddsd_matmul_info.cc183 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Dgatherd_info.cc198 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Dbroadcast_to_info.cc132 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Dtile_info.cc204 Shapes tmp_inputs_shape = {full_multiples_}; in GenerateOpStrategies() local
Dlayer_norm_info.cc218 Shapes tmp_inputs_shape = {input_shape_}; in GenerateOpStrategies() local
Dmatmul_dds_info.cc291 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Doperator_info.cc1107 Shapes tmp_inputs_shape = {inputs_shape[1], inputs_shape[1]}; in GenerateStrategiesForBroadcastLeft() local
1156 Shapes tmp_inputs_shape = {inputs_shape[0], inputs_shape[0]}; in GenerateStrategiesForBroadcastRight() local
1218 Shapes tmp_inputs_shape = {max_shape, max_shape}; in GenerateStrategiesForBroadcastBoth() local