Searched refs:dev_mat_shape (Results 1 – 1 of 1) sorted by relevance
383 auto dev_mat_shape = tensor_layout->device_arrangement().array(); in IsFullySplitParameter() local387 DeviceMatrix dev_matrix(rank, rank_list, dev_mat_shape); in IsFullySplitParameter()