Searched refs:split_nodes_for_dimension (Results 1 – 1 of 1) sorted by relevance
825 std::queue<Node*> split_nodes_for_dimension; in CreateOrGetSplitNodesForInputSharding() local845 split_nodes_for_dimension.emplace(root_split_node); in CreateOrGetSplitNodesForInputSharding()851 int num_split_nodes_in_dimension = split_nodes_for_dimension.size(); in CreateOrGetSplitNodesForInputSharding()853 Node* input_split_node = split_nodes_for_dimension.front(); in CreateOrGetSplitNodesForInputSharding()854 split_nodes_for_dimension.pop(); in CreateOrGetSplitNodesForInputSharding()867 split_nodes_for_dimension.emplace(split_node); in CreateOrGetSplitNodesForInputSharding()880 while (!split_nodes_for_dimension.empty()) { in CreateOrGetSplitNodesForInputSharding()881 Node* split_node = split_nodes_for_dimension.front(); in CreateOrGetSplitNodesForInputSharding()882 split_nodes_for_dimension.pop(); in CreateOrGetSplitNodesForInputSharding()