Searched refs:expand_partition_groups (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.cc | 1598 auto expand_partition_groups = in GetPerGroupCollectiveOpsCreator() local
1619 [creator, expand_partition_groups]( in GetPerGroupCollectiveOpsCreator()
1624 b, operand, reduction, expand_partition_groups(partition_subgroups), in GetPerGroupCollectiveOpsCreator()
1646 [creator, expand_partition_groups]( in GetPerGroupCollectiveOpsCreator()
1651 b, operands, expand_partition_groups(partition_subgroups), in GetPerGroupCollectiveOpsCreator()
1656 [creator, expand_partition_groups]( in GetPerGroupCollectiveOpsCreator()
1662 expand_partition_groups(partition_subgroups), channel_id, in GetPerGroupCollectiveOpsCreator()
|