Searched defs:partition_id (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | fft_handler.cc | 55 int64* next_channel_id, HloInstruction* partition_id, SpmdBuilder* b) { in PadEachPartitionWithHaloExchange()
173 HloInstruction* partition_id, in GetCorrectionFactor()
230 int64 num_partitions, HloInstruction* partition_id, int64* next_channel_id, in GetFinalFftUsingCollectivePermute()
|
| D | convolution_handler.cc | 234 HloInstruction* partition_id, HloModule* module, SpmdBuilder* b) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
521 HloInstruction* partition_id, HloModule* module, SpmdBuilder* b) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
824 HloInstruction* partition_id, HloModule* module, SpmdBuilder* b) { in PartitionConvolutionBaseCase()
972 HloInstruction* partition_id, HloModule* module, SpmdBuilder* b) { in PartitionConvolution()
|
| D | spmd_partitioner_util.cc | 168 int64 partition_id) { in MakeNonPaddedShapeForGivenPartition()
206 HloInstruction* partition_id, SpmdBuilder* b, in MakePartitionOffsets()
240 const HloSharding& sharding, HloInstruction* partition_id, SpmdBuilder* b) { in MakeTiledPartitionOrdinals()
413 int64* next_channel_id, HloInstruction* partition_id, SpmdBuilder* b) { in TileToPartialReplicateHaloExchange()
507 int64* next_channel_id, HloInstruction* partition_id, SpmdBuilder* b) { in PadFromPartialReplicateShape()
1573 HloInstruction* partition_id, in GetInGroupPartitionId()
1694 HloInstruction* replicated, HloInstruction* partition_id, in PerGroupSliceFromReplicated()
|
| D | gather_scatter_handler.cc | 55 HloInstruction* partition_id, absl::Span<const int64> index_map, in IndexBoundsForGatherScatterOperandPartitionedOnTrivialSliceDims()
579 auto partition_id = partition_id_; in HandleScatter() local
|
| D | spmd_partitioner.h | 287 HloInstruction* partition_id; member
|
| D | dot_handler.cc | 666 auto partition_id = in PartitionBaseCase() local
1058 auto partition_id = in PartitionBaseCase() local
|
| D | spmd_partitioner_test.cc | 4295 auto partition_id = in TEST_F() local
|
| /external/webrtc/modules/video_coding/ |
| D | session_info.cc | 353 const int partition_id = in FindPartitionEnd() local
|
| /external/libvpx/libvpx/vpx/ |
| D | vpx_encoder.h | 173 int partition_id; member
|
| /external/libaom/libaom/aom/ |
| D | aom_encoder.h | 145 int partition_id; member
|