Searched refs:dev_max (Results 1 – 1 of 1) sorted by relevance
420 Shape dev_max = device_arrangement_.array(); in GenerateOptShardSliceShape() local423 for (size_t i = 0; i < dev_max.size(); i++) { in GenerateOptShardSliceShape()425 repeated_dev.push_back(dev_max[dev_max.size() - 1 - i]); in GenerateOptShardSliceShape()426 dev_max[dev_max.size() - 1 - i] = 1; in GenerateOptShardSliceShape()443 split_num = dev_max[dev_max.size() - 1 - static_cast<size_t>(tensor_map[0])] * repeated_num; in GenerateOptShardSliceShape()