Searched refs:node_to_split_dim (Results 1 – 1 of 1) sorted by relevance
826 absl::flat_hash_map<Node*, int> node_to_split_dim; in CreateOrGetSplitNodesForInputSharding() local846 node_to_split_dim[root_split_node] = split_dimension; in CreateOrGetSplitNodesForInputSharding()868 node_to_split_dim[split_node] = split_dimension; in CreateOrGetSplitNodesForInputSharding()