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