Home
last modified time | relevance | path

Searched defs:out_dev_matrix_shape (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dself_define_shard_info.cc114 …auto out_dev_matrix_shape = outputs_tensor_info_[0].tensor_layout().device_arrangement_origin().ar… in InferAsLossDivisorByLayout() local
Drms_norm_info.cc154 …auto out_dev_matrix_shape = outputs_tensor_info_[0].tensor_layout().device_arrangement_origin().ar… in InferAsLossDivisorByLayout() local
Dlayer_norm_info.cc230 …auto out_dev_matrix_shape = outputs_tensor_info_[0].tensor_layout().device_arrangement_origin().ar… in InferAsLossDivisorByLayout() local
Dgather_info.h85 Shape out_dev_matrix_shape() const { return out_dev_matrix_shape_; } in out_dev_matrix_shape() function
Dflash_attention_score_info.cc589 …auto out_dev_matrix_shape = attention_out_tensor_info.tensor_layout().device_arrangement_origin().… in InferAsLossDivisorByLayout() local
Doperator_info.cc2529 …auto out_dev_matrix_shape = outputs_tensor_info_[0].tensor_layout().device_arrangement_origin().ar… in InferAsLossDivisorByLayout() local