Searched refs:CreateVisitor (Results 1 – 2 of 2) sorted by relevance
226 virtual std::unique_ptr<SpmdPartitioningVisitor> CreateVisitor(
3636 CreateVisitor(computation, num_partitions_, num_replicas_, in PartitionComputation()3641 std::unique_ptr<SpmdPartitioningVisitor> SpmdPartitioner::CreateVisitor( in CreateVisitor() function in xla::spmd::SpmdPartitioner