Searched refs:temp_dev_matrix (Results 1 – 1 of 1) sorted by relevance
589 DeviceMatrix temp_dev_matrix(rank, stage_device_list_, dev_mat); in CreateGroupForOptShard() local591 if (temp_dev_matrix.GetDevicesAlongDim(0, &mirror_group_devices) != SUCCESS) { in CreateGroupForOptShard()