Home
last modified time | relevance | path

Searched refs:create_partition_id (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h115 std::function<HloInstruction*(SpmdBuilder*)> create_partition_id; member
Dspmd_partitioner_util.cc1594 result.create_partition_id = [creator, device_groups](SpmdBuilder* b) { in GetPerGroupCollectiveOpsCreator()
1595 return GetInGroupPartitionId(creator.create_partition_id(b), device_groups, in GetPerGroupCollectiveOpsCreator()
Ddot_handler.cc667 lhs.state().collective_ops_creator.create_partition_id(&body_b); in PartitionBaseCase()
1059 lhs.state().collective_ops_creator.create_partition_id(&body_b); in PartitionBaseCase()
Dspmd_partitioner.cc1363 partition_id_(collective_ops_creator_.create_partition_id(&b_)), in SpmdPartitioningVisitor()