Searched refs:partition_array_dims (Results 1 – 1 of 1) sorted by relevance
199 const auto& partition_array_dims = partition_array.shape().dims(); in Run() local206 {num_partitions * partition_array_dims[0]}); in Run()210 for (int i = 0; i < num_partitions * partition_array_dims[0]; ++i) { in Run()212 perm_data[i] = p * partition_array_dims[0] + i / num_partitions; in Run()