Home
last modified time | relevance | path

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

12

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dtopk_info.cc208 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
229 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Dquant_info.cc85 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
186 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
413 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Dgathernd_info.cc131 Shapes tmp_inputs_shape = {inputs_shape_[1]}; in GenerateOpStrategies() local
Dscatter_ops_info.cc149 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Dselect_info.cc131 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Ddynamic_quant_ext_info.cc128 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Dpad_info.cc169 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Ddsd_matmul_info.cc165 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Dconcat_info.cc177 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Dbroadcast_to_info.cc134 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Dpack_info.cc163 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Dbias_add_info.cc100 Shapes tmp_inputs_shape = {inputs_shape_[0], inputs_shape_[0]}; in GenerateOpStrategies() local
Dsplit_info.cc211 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Dbatchnorm_info.cc350 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Dtile_info.cc287 Shapes tmp_inputs_shape = {full_multiples_}; in GenerateOpStrategies() local
Dmatmul_dds_info.cc296 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Dgatherd_info.cc233 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Dscatter_math_ops_info.cc242 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Drms_norm_info.cc215 Shapes tmp_inputs_shape = {input_shape_}; in GenerateOpStrategies() local
Dscatter_nd_ops_info.cc346 Shapes tmp_inputs_shape = {inputs_shape_[0]}; in GenerateOpStrategies() local
Dlayer_norm_info.cc295 Shapes tmp_inputs_shape = {input_shape_}; in GenerateOpStrategies() local
Dweight_quant_batch_matmul_info.cc565 Shapes tmp_inputs_shape = {tmp_shape}; in GenerateOpStrategies() local
Dquant_batch_matmul_info.cc564 Shapes tmp_inputs_shape = {tmp_shape}; in GenerateOpStrategies() local
Doperator_info.cc1809 Shapes tmp_inputs_shape = {inputs_shape[1], inputs_shape[1]}; in GenerateStrategiesForBroadcastLeft() local
1858 Shapes tmp_inputs_shape = {inputs_shape[0], inputs_shape[0]}; in GenerateStrategiesForBroadcastRight() local
1920 Shapes tmp_inputs_shape = {max_shape, max_shape}; in GenerateStrategiesForBroadcastBoth() local
2099 Shapes tmp_inputs_shape = {tmp_input_shape}; in GenerateStrategiesForDependentInputs() local
Dmax_avg_pool_3d_info.cc966 Shapes tmp_inputs_shape = inputs_shape_; in GenerateOpStrategies() local

12