| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | collective_ops_utils.cc | 168 for (int partition_id = 0; partition_id < partition_count; in GetParticipatingDevicesGroups() local
190 for (int partition_id : replica_group.replica_ids()) { in GetParticipatingDevicesGroups() local
208 for (int partition_id = 0; partition_id < partition_count; in GetParticipatingDevicesGroups() local
226 int partition_id = flattened_id % partition_count; in GetParticipatingDevicesGroups() local
276 for (int partition_id : participating_partitions) { in GetParticipatingDevices() local
291 for (int partition_id = 0; partition_id < partition_count; in GetParticipatingDevices() local
320 int partition_id = flattened_id % partition_count; in GetParticipatingDevices() local
|
| D | collective_ops_utils_test.cc | 379 for (int64_t partition_id = 0; partition_id < num_partitions; in TEST_P() local
|
| D | reduce_scatter_utils.cc | 324 int64_t partition_id = global_id % num_partitions; in MatchReduceScatter() local
|
| /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_t num_partitions, HloInstruction* partition_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_t 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()
1620 HloInstruction* partition_id, in GetInGroupPartitionId()
1741 HloInstruction* replicated, HloInstruction* partition_id, in PerGroupSliceFromReplicated()
|
| D | gather_scatter_handler.cc | 55 HloInstruction* partition_id, absl::Span<const int64> index_map, in IndexBoundsForGatherScatterOperandPartitionedOnTrivialSliceDims()
598 auto partition_id = MakePartitioningState().partition_id; in HandleScatter() local
|
| D | spmd_partitioner.h | 312 HloInstruction* partition_id; member
|
| D | dot_handler.cc | 784 auto partition_id = in PartitionBaseCase() local
1157 auto partition_id = in PartitionBaseCase() local
|
| D | spmd_partitioner_test.cc | 4638 auto partition_id = in TEST_F() local
|
| /external/rmi4utils/rmi4update/ |
| D | rmi4update.cpp | 341 unsigned char partition_id = FLASH_CONFIG_PARTITION; in ReadFlashConfig() local
646 unsigned char partition_id; in WriteFirmwareV7() local
757 unsigned char partition_id; in WriteCoreConfigV7() local
866 unsigned char partition_id; in WriteFlashConfigV7() local
|
| D | rmi4update.h | 143 unsigned short partition_id; member
|
| /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 | 175 int partition_id; member
|
| /external/libaom/libaom/aom/ |
| D | aom_encoder.h | 125 int partition_id; member
|
| /external/bcc/libbpf-tools/arm64/ |
| D | vmlinux.h | 59133 void *partition_id; member
64753 u8 partition_id; member
70954 guid_t partition_id; member
|
| D | vmlinux_510.h | 59133 void *partition_id; member
64753 u8 partition_id; member
70954 guid_t partition_id; member
|
| /external/bcc/libbpf-tools/powerpc/ |
| D | vmlinux_510.h | 92788 u8 partition_id; member
|
| D | vmlinux.h | 92788 u8 partition_id; member
|
| /external/libbpf/.github/actions/build-selftests/ |
| D | vmlinux.h | 60787 u8 partition_id; member
|
| /external/bcc/libbpf-tools/x86/ |
| D | vmlinux.h | 61784 u8 partition_id; member
|
| D | vmlinux_505.h | 61784 u8 partition_id; member
|