Home
last modified time | relevance | path

Searched defs:dev_shape (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Ddevice_matrix.cc32 DeviceMatrix::DeviceMatrix(int64_t rank, RankList dev_list, Shape dev_shape) in DeviceMatrix()
103 Shape ConvertRankToCoordinate(int64_t rank, const Shape &dev_shape) { in ConvertRankToCoordinate()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Darithmetic_info.cc100 Shape dev_shape; in InferDevMatrixShape() local
164 Shape dev_shape; in InferTensorMap() local