Searched refs:num_split_nodes_in_dimension (Results 1 – 1 of 1) sorted by relevance
851 int num_split_nodes_in_dimension = split_nodes_for_dimension.size(); in CreateOrGetSplitNodesForInputSharding() local852 for (int i = 0; i < num_split_nodes_in_dimension; ++i) { in CreateOrGetSplitNodesForInputSharding()