Home
last modified time | relevance | path

Searched defs:partition_id (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcollective_ops_utils.cc169 for (int partition_id = 0; partition_id < partition_count; in GetParticipatingDevicesGroups() local
191 for (int partition_id : replica_group.replica_ids()) { in GetParticipatingDevicesGroups() local
209 for (int partition_id = 0; partition_id < partition_count; in GetParticipatingDevicesGroups() local
227 int partition_id = flattened_id % partition_count; in GetParticipatingDevicesGroups() local
277 for (int partition_id : participating_partitions) { in GetParticipatingDevices() local
292 for (int partition_id = 0; partition_id < partition_count; in GetParticipatingDevices() local
321 int partition_id = flattened_id % partition_count; in GetParticipatingDevices() local
Dcollective_ops_utils_test.cc379 for (int64_t partition_id = 0; partition_id < num_partitions; in TEST_P() local
Dreduce_scatter_utils.cc342 int64_t partition_id = global_id % num_partitions; in MatchReduceScatter() local
/external/rmi4utils/rmi4update/
Drmi4update.cpp456 unsigned char partition_id = FLASH_CONFIG_PARTITION; in ReadFlashConfig() local
788 unsigned char partition_id; in WriteFirmwareV7() local
920 unsigned char partition_id; in WriteCoreConfigV7() local
1048 unsigned char partition_id; in WriteFlashConfigV7() local
1180 unsigned char partition_id; in WriteFLDV7() local
1318 unsigned char partition_id; in WriteGlobalParametersV7() local
Drmi4update.h149 unsigned short partition_id; member
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dfft_handler.cc53 int64_t* next_channel_id, HloInstruction* partition_id, SpmdBuilder* b) { in PadEachPartitionWithHaloExchange()
171 HloInstruction* partition_id, in GetCorrectionFactor()
228 int64_t num_partitions, HloInstruction* partition_id, in GetFinalFftUsingCollectivePermute()
Dconvolution_handler.cc234 HloInstruction* partition_id, HloModule* module, SpmdBuilder* b) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
523 HloInstruction* partition_id, HloModule* module, SpmdBuilder* b) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
826 HloInstruction* partition_id, HloModule* module, SpmdBuilder* b) { in PartitionConvolutionBaseCase()
974 HloInstruction* partition_id, HloModule* module, SpmdBuilder* b) { in PartitionConvolution()
Dspmd_partitioner_util.cc117 int64_t partition_id) { in MakeNonPaddedShapeForGivenPartition()
157 HloInstruction* partition_id, SpmdBuilder* b, in MakePartitionOffsets()
191 const HloSharding& sharding, HloInstruction* partition_id, SpmdBuilder* b) { in MakeTiledPartitionOrdinals()
328 int64_t* next_channel_id, HloInstruction* partition_id, SpmdBuilder* b) { in TileToPartialReplicateHaloExchange()
422 int64_t* next_channel_id, HloInstruction* partition_id, SpmdBuilder* b) { in PadFromPartialReplicateShape()
1459 HloInstruction* partition_id, in GetInGroupPartitionId()
1580 HloInstruction* replicated, HloInstruction* partition_id, in PerGroupSliceFromReplicated()
Dgather_scatter_handler.cc88 HloInstruction* partition_id, absl::Span<const int64_t> index_map, in IndexBoundsForGatherScatterOperandPartitionedOnTrivialSliceDims()
909 auto partition_id = visitor->MakePartitioningState().partition_id; in PartitionScatterIndexPassthroughPartition() local
Dspmd_partitioner.h327 HloInstruction* partition_id; member
Ddot_handler.cc1042 auto partition_id = in PartitionBaseCase() local
1419 auto partition_id = in PartitionBaseCase() local
Dspmd_partitioner_test.cc6490 auto partition_id = in TEST_F() local
/external/webrtc/modules/video_coding/
Dsession_info.cc355 const int partition_id = in FindPartitionEnd() local
/external/libvpx/vpx/
Dvpx_encoder.h176 int partition_id; member
/external/libaom/aom/
Daom_encoder.h134 int partition_id; member
/external/openthread/tests/scripts/thread-cert/
Dmle.py424 partition_id, argument
437 def partition_id(self): member in LeaderData
Dnode.py1559 def set_preferred_partition_id(self, partition_id): argument
/external/bcc/libbpf-tools/arm64/
Dvmlinux.h59133 void *partition_id; member
64753 u8 partition_id; member
70954 guid_t partition_id; member
Dvmlinux_510.h59133 void *partition_id; member
64753 u8 partition_id; member
70954 guid_t partition_id; member
/external/libbpf/.github/actions/build-selftests/
Dvmlinux.h60787 u8 partition_id; member
/external/bcc/libbpf-tools/powerpc/
Dvmlinux_510.h92788 u8 partition_id; member
Dvmlinux.h92788 u8 partition_id; member
/external/bcc/libbpf-tools/x86/
Dvmlinux_505.h61784 u8 partition_id; member
Dvmlinux.h61784 u8 partition_id; member