Searched refs:collective_ops_creator_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.h | 192 collective_ops_creator_(std::move(collective_ops_creator)) {} in SpmdPartitioner()
250 SPMDCollectiveOpsCreator collective_ops_creator_; variable
488 state.collective_ops_creator = collective_ops_creator_; in MakePartitioningState()
527 SPMDCollectiveOpsCreator collective_ops_creator_; variable
|
| D | spmd_partitioner.cc | 1360 collective_ops_creator_(collective_ops_creator), in SpmdPartitioningVisitor()
1363 partition_id_(collective_ops_creator_.create_partition_id(&b_)), in SpmdPartitioningVisitor()
2768 hlo->dimensions(), collective_ops_creator_, hlo->to_apply()); in HandleReduce()
3290 partition_ordinals[i], collective_ops_creator_, next_channel_id_, &b_); in HandleSelectAndScatter()
3307 partition_ordinals[i], collective_ops_creator_, next_channel_id_, &b_); in HandleSelectAndScatter()
3637 collective_ops_creator_, next_channel_id, logger, options_); in PartitionComputation()
|
| D | gather_scatter_handler.cc | 607 all_dims, collective_ops_creator_, scatter->to_apply()); in HandleScatter()
|