Searched refs:partitioner_ (Results 1 – 2 of 2) sorted by relevance
491 state.partitioner = partitioner_; in MakePartitioningState()546 SpmdPartitioner* partitioner_; variable
1366 partitioner_(partitioner) {} in SpmdPartitioningVisitor()2766 reduce = partitioner_->AllReduceAlongShardingDims( in HandleReduce()2845 TF_RETURN_IF_ERROR(partitioner_ in HandleWhile()2850 TF_RETURN_IF_ERROR(partitioner_ in HandleWhile()2879 TF_RETURN_IF_ERROR(partitioner_ in HandleConditional()