Searched refs:CreateSplitNode (Results 1 – 1 of 1) sorted by relevance
712 xla::StatusOr<Node*> CreateSplitNode(const int num_splits, const int dim, in CreateSplitNode() function836 CreateSplitNode( in CreateOrGetSplitNodesForInputSharding()860 CreateSplitNode( in CreateOrGetSplitNodesForInputSharding()