Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/
Dshard.py61 def __init__(self, device_matrix, alias_name): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dgrouped_matmul_info.cc255 TensorInfoBasePtr GroupedMatmulInfo::CreateTensorInfo(const Shape &device_matrix, const ShapeBasePt… in CreateTensorInfo()
Dmatmul_info.cc796 …auto device_matrix = DeviceMatrix(g_device_manager->global_rank(), g_device_manager->GetDeviceList… in InferForwardCommunicationByLayout() local
1117 …auto device_matrix = DeviceMatrix(g_device_manager->global_rank(), g_device_manager->GetDeviceList… in ComputeNDTPReplaceGraph() local
1273 …auto device_matrix = DeviceMatrix(g_device_manager->global_rank(), g_device_manager->GetDeviceList… in ComputePostMatMulGraph() local
Doperator_info.cc550 TensorInfoBasePtr CreateTensorInfo(const Shape &device_matrix, const ShapeBasePtr &inputs_shape, in CreateTensorInfo()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dtensor_redistribution.cc51 auto device_matrix = from_layout.device_arrangement_origin().array(); in MakeFromToLayout() local
73 auto device_matrix = to_layout.device_arrangement_origin().array(); in MakeFromToLayout() local
Dtensor_layout.cc111 Status TensorLayout::InitFromExtendVector(const Shape &device_matrix, const std::vector<Shape> &ten… in InitFromExtendVector()