Home
last modified time | relevance | path

Searched defs:next_channel_id (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dfft_handler.cc55 int64* next_channel_id, HloInstruction* partition_id, SpmdBuilder* b) { in PadEachPartitionWithHaloExchange()
162 int64* next_channel_id, SpmdBuilder* b) { in ShuffleDataWithAllToAll()
230 int64 num_partitions, HloInstruction* partition_id, int64* next_channel_id, in GetFinalFftUsingCollectivePermute()
Dspmd_partitioner_util.cc413 int64* next_channel_id, HloInstruction* partition_id, SpmdBuilder* b) { in TileToPartialReplicateHaloExchange()
507 int64* next_channel_id, HloInstruction* partition_id, SpmdBuilder* b) { in PadFromPartialReplicateShape()
803 int64* next_channel_id, SpmdBuilder* b) { in ExchangeHalo()
905 int64* next_channel_id, SpmdBuilder* b) { in ExchangeHalo()
931 int64* next_channel_id, SpmdBuilder* b, bool mask_invalid_region) { in ExchangeHaloAndGetValidData()
1631 int64 next_channel_id) { in GetPerGroupCollectiveOpsCreator()
Dspmd_partitioner.cc1354 int64* next_channel_id, SpmdLogger* logger, SpmdPartitionerOptions options, in SpmdPartitioningVisitor()
3501 int64* next_channel_id, absl::Span<const int64> selected_dims, in AllGatherShards()
3510 int64* next_channel_id, absl::Span<const int64> selected_dims, in AllGatherShardsInternal()
3600 int64* next_channel_id, absl::Span<const int64> selected_dims, in AllReduceAlongShardingDims()
3610 int64* next_channel_id, absl::Span<const int64> selected_dims, in AllReduceAlongShardingDimsInternal()
3634 int64* next_channel_id, SpmdLogger* logger) { in PartitionComputation()
3644 int64* next_channel_id, SpmdLogger* logger, in CreateVisitor()
3674 int64 next_channel_id = hlo_query::NextChannelId(*module); in Run() local
Dspmd_partitioner.h289 int64* next_channel_id; member
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_query.cc137 int64 next_channel_id = 1; in NextChannelId() local