Home
last modified time | relevance | path

Searched defs:outputs_tensor_map (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dself_define_shard_info.cc106 TensorMaps outputs_tensor_map = outputs_tensor_info_[0].tensor_layout().tensor_map_before(); in InferAsLossDivisorByLayout() local
Drms_norm_info.cc147 TensorMaps outputs_tensor_map = outputs_tensor_info_[0].tensor_layout().tensor_map_before(); in InferAsLossDivisorByLayout() local
Dlayer_norm_info.cc223 TensorMaps outputs_tensor_map = outputs_tensor_info_[0].tensor_layout().tensor_map_before(); in InferAsLossDivisorByLayout() local
Dgather_info.h76 TensorMaps outputs_tensor_map() const { return outputs_tensor_map_; } in outputs_tensor_map() function
Doperator_info.cc2521 TensorMaps outputs_tensor_map = outputs_tensor_info_[0].tensor_layout().tensor_map_before(); in InferAsLossDivisorByLayout() local