| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | fft_handler.cc | 55 int64* next_channel_id, HloInstruction* partition_id, SpmdBuilder* b) { in PadEachPartitionWithHaloExchange() argument
102 MakeTiledPartitionOrdinals(sharding, partition_id, b); in PadEachPartitionWithHaloExchange()
173 HloInstruction* partition_id, in GetCorrectionFactor() argument
192 ShapeUtil::ChangeElementType(partition_id->shape(), in GetCorrectionFactor()
194 partition_id)); in GetCorrectionFactor()
230 int64 num_partitions, HloInstruction* partition_id, int64* next_channel_id, in GetFinalFftUsingCollectivePermute() argument
235 ShapeUtil::ChangeElementType(partition_id->shape(), in GetFinalFftUsingCollectivePermute()
237 partition_id)); in GetFinalFftUsingCollectivePermute()
387 partitioned_input.state().partition_id, partitioned_input.state().b); in HandleFft()
416 result, num_partitions_, partitioned_input.state().partition_id, in HandleFft()
[all …]
|
| D | spmd_partitioner_util.h | 99 int64 partition_id);
107 HloInstruction* partition_id, SpmdBuilder* b,
112 const HloSharding& sharding, HloInstruction* partition_id, SpmdBuilder* b);
353 HloInstruction* replicated, HloInstruction* partition_id,
371 int64* next_channel_id, HloInstruction* partition_id, SpmdBuilder* b);
394 int64* next_channel_id, HloInstruction* partition_id, SpmdBuilder* b);
|
| D | gather_scatter_handler.cc | 55 HloInstruction* partition_id, absl::Span<const int64> index_map, in IndexBoundsForGatherScatterOperandPartitionedOnTrivialSliceDims() argument
58 operand.base_shape(), operand.sharding(), partition_id, b); in IndexBoundsForGatherScatterOperandPartitionedOnTrivialSliceDims()
237 operand, indices, operand.state().partition_id, start_index_map, in ParititonTrivialIndexedOperandDimension()
376 operand.base_shape(), operand_sharding, operand.state().partition_id, in PartitionIndexParallelDimensions()
579 auto partition_id = partition_id_; in HandleScatter() local
586 partition_id = per_group_partitioner_state.partition_id; in HandleScatter()
591 ShapeUtil::MakeScalarShape(PRED), partition_id)); in HandleScatter()
|
| D | spmd_partitioner_util.cc | 168 int64 partition_id) { in MakeNonPaddedShapeForGivenPartition() argument
174 sharding.GetSubSharding(shape, {i}), partition_id)); in MakeNonPaddedShapeForGivenPartition()
183 if (partition_id == *sharding.UniqueDevice()) { in MakeNonPaddedShapeForGivenPartition()
191 sharding.TileOffsetForDevice(shape, partition_id); in MakeNonPaddedShapeForGivenPartition()
193 sharding.TileLimitForDevice(shape, partition_id); in MakeNonPaddedShapeForGivenPartition()
195 if (sharding.UsesDevice(partition_id)) { in MakeNonPaddedShapeForGivenPartition()
206 HloInstruction* partition_id, SpmdBuilder* b, in MakePartitionOffsets() argument
231 ShapeUtil::MakeShape(S32, {1}), offset_table, {partition_id}, {1})); in MakePartitionOffsets()
240 const HloSharding& sharding, HloInstruction* partition_id, SpmdBuilder* b) { in MakeTiledPartitionOrdinals() argument
247 return MakePartitionOffsets(table_shape, sharding, partition_id, b); in MakeTiledPartitionOrdinals()
[all …]
|
| D | convolution_handler.cc | 234 HloInstruction* partition_id, HloModule* module, SpmdBuilder* b) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() argument
349 MakeTiledPartitionOrdinals(lhs.sharding(), partition_id, b); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
521 HloInstruction* partition_id, HloModule* module, SpmdBuilder* b) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() argument
656 MakeTiledPartitionOrdinals(lhs.sharding(), partition_id, b); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
824 HloInstruction* partition_id, HloModule* module, SpmdBuilder* b) { in PartitionConvolutionBaseCase() argument
862 conv_window, original_hlo, partition_id, module, b)); in PartitionConvolutionBaseCase()
871 conv_window, original_hlo, partition_id, module, b)); in PartitionConvolutionBaseCase()
972 HloInstruction* partition_id, HloModule* module, SpmdBuilder* b) { in PartitionConvolution() argument
979 num_partitions, options, partition_id, module, b)); in PartitionConvolution()
|
| D | convolution_handler.h | 37 HloInstruction* partition_id, HloModule* module, SpmdBuilder* b);
|
| D | spmd_partitioner.h | 287 HloInstruction* partition_id; member
487 state.partition_id = partition_id_; in MakePartitioningState()
|
| D | spmd_partitioner.cc | 461 hlo_, state_.partition_id, target_grouped.device_groups, group_dims, in ReshardNoCache()
473 MakePartitionOffsets(shape, target, state_.partition_id, state_.b), in ReshardNoCache()
518 state_.partition_id, state_.b); in PadWithValue()
567 MakeTiledPartitionOrdinals(target, state_.partition_id, state_.b); in ReshardAsWindowedInput()
889 state_.partition_id, state_.b, dims); in ReplicatePartial()
950 partitioned_hlo.state().partition_id, partitioned_hlo.state().b); in ReshardToPartialReplicateWithAllGather()
1019 state_.partition_id, state_.b); in ReshardFromPartialReplicateWithDynamicSlice()
1034 state_.partition_id, state_.b); in ReshardFromPartialReplicateWithDynamicSlice()
1036 state_.partition_id, state_.b); in ReshardFromPartialReplicateWithDynamicSlice()
1067 ShapeUtil::MakeShape(PRED, {}), state_.partition_id, src_core_id, in Broadcast()
[all …]
|
| D | dot_handler.cc | 666 auto partition_id = in PartitionBaseCase() local
673 i->shape(), HloOpcode::kAdd, i, partition_id)); in PartitionBaseCase()
676 i->shape(), HloOpcode::kAdd, partition_count, partition_id)); in PartitionBaseCase()
829 state.partition_id = data_partition_id; in PartitionBaseCase()
1058 auto partition_id = in PartitionBaseCase() local
1062 i->shape(), HloOpcode::kAdd, i, partition_id)); in PartitionBaseCase()
1081 state.partition_id = data_partition_id; in PartitionBaseCase()
1755 operand.Replicate().hlo(), operand.state().partition_id, in PartitionDotGroupOnBatch()
2528 lhs.state().partition_id, module, b)); in PartitionDot()
|
| D | spmd_partitioner_test.cc | 4295 auto partition_id = in TEST_F() local
4300 op::Broadcast(op::Compare(partition_id, op::Constant())), in TEST_F()
|
| /external/libaom/libaom/aom/ |
| D | aom_encoder.h | 145 int partition_id; member
|
| /external/libvpx/libvpx/vpx/ |
| D | vpx_encoder.h | 173 int partition_id; member
|
| /external/webrtc/modules/video_coding/ |
| D | session_info.cc | 353 const int partition_id = in FindPartitionEnd() local
365 (beginning && current_partition_id != partition_id)) { in FindPartitionEnd()
|
| /external/libvpx/libvpx/vp8/ |
| D | vp8_cx_iface.c | 1008 pkt.data.frame.partition_id = i; in vp8e_encode()
1026 pkt.data.frame.partition_id = -1; in vp8e_encode()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | lhlo_ops.td | 625 def LHLO_PartitionIdOp : LHLO_Op<"partition_id", []>, BASE_HLO_PartitionIdOp {
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo.proto | 183 // a linearized id of `replica_id * partition_count + partition_id`.
|
| /external/libvpx/libvpx/ |
| D | vpxenc.c | 1415 if (pkt->data.frame.partition_id <= 0) { in get_cx_data()
|
| /external/libvpx/libvpx/vp9/ |
| D | vp9_cx_iface.c | 1334 pkt.data.frame.partition_id = -1; in encoder_encode()
|
| /external/cpuinfo/test/dmesg/ |
| D | galaxy-j7-uae.log | 690 [ 16.975176] [7: vold: 2362] ecryptfs_parse_options : partition_id : 0
693 [ 16.979832] [7: vold: 2362] ecryptfs_parse_options : partition_id : 1
|
| D | galaxy-s8-us.log | 5247 [ 17.157454] [2: UsbPortManager: 1837] ecryptfs_parse_options : partition_id : 0
5311 [ 17.225103] [2:WifiStateMachin: 1794] ecryptfs_parse_options : partition_id : 1
|
| D | galaxy-a8-2018.log | 5972 [ 15.978062] [6: kworker/u16:4: 1586] ecryptfs_parse_options : partition_id : 0
5979 [ 15.985790] [6: kworker/u16:4: 1586] ecryptfs_parse_options : partition_id : 1
|
| /external/libaom/libaom/apps/ |
| D | aomenc.c | 2073 if (pkt->data.frame.partition_id <= 0) { in get_cx_data()
|
| /external/libaom/libaom/av1/ |
| D | av1_cx_iface.c | 2276 pkt.data.frame.partition_id = -1; in encoder_encode()
|