Searched refs:DoPartition (Results 1 – 2 of 2) sorted by relevance
497 StatusOr<bool> DoPartition(HloComputation* computation,
3381 StatusOr<bool> SpmdPartitioningVisitor::DoPartition( in DoPartition() function in xla::spmd::SpmdPartitioningVisitor3638 return visitor->DoPartition(computation, root_sharding, options_); in PartitionComputation()