Home
last modified time | relevance | path

Searched refs:set_dimensions (Results 1 – 25 of 38) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_reduce_scatter_creator.cc68 scatter_shape.set_dimensions(split_dim, in Run()
76 scatter_shape.set_dimensions(split_dim, scatter_dim_size); in Run()
Dcudnn_pad_for_convolutions.cc233 new_input_shape->set_dimensions(dnums.input_feature_dimension(), 4); in TryResolvePaddedShapesForTensorCore()
234 new_filter_shape->set_dimensions(dnums.kernel_input_feature_dimension(), 4); in TryResolvePaddedShapesForTensorCore()
237 s->set_dimensions(dim, RoundUpTo<int64_t>(s->dimensions(dim), 8)); in TryResolvePaddedShapesForTensorCore()
360 s->set_dimensions( in TryResolvePaddedShapesForIntegerConvolution()
Dvariadic_op_splitter.cc66 new_shape.set_dimensions(concat->concatenate_dimension(), in SplitConcatenate()
Dcublas_pad_for_gemms.cc43 s.set_dimensions(dim, in PadForGemm()
Dgpu_conv_rewriter.cc550 new_shape.set_dimensions(output_feature_dimension, in MatchBackwardInput()
646 new_shape.set_dimensions(input_feature_dimension, input_feature * num_groups); in ConvertBatchGroupedToFeatureGroupedConvolution()
Dbuffer_comparator.cc796 shape_.set_dimensions(0, prev_zero_dim_size * 2); in BufferComparator()
Dgpu_conv_padding_legalization.cc324 new_backward_conv_shape.set_dimensions( in CanonicalizeBackwardInputConvolution()
/external/tensorflow/tensorflow/compiler/xla/service/
Dall_gather_broadcast_reorder.cc104 new_ag_shape.set_dimensions(ag_dim_index, in Run()
152 shape.set_dimensions(0, ag_factor); in Run()
Dall_to_all_decomposer.cc101 slice_shape.set_dimensions(split_dim, split_size); in ExpandInstruction()
Dconvolution_group_converter.cc111 expanded_shape.set_dimensions( in ExpandedFilterShape()
358 reduce_window_shape.set_dimensions(output_batch_dimension, 1); in HandleBatchGroupCount()
Dwhile_loop_concat_code_motion.cc101 shape.set_dimensions(concat_dim, dim_size); in GetConcatShape()
109 shape.set_dimensions(concat_dim, element_sizes[element_index]); in CreateSlice()
898 data_shape.set_dimensions( in RewriteLoopWithConcatGroups()
Dhlo_sharding.cc756 result_shape.set_dimensions( in TileShape()
776 result_shape.set_dimensions(i, limit - offset); in TileShape()
Dshape_inference.cc892 output_shape.set_dimensions(dimension_to_match, small_dimension_size); in InferInDimBroadcastShape()
1919 result.set_dimensions(result.dimensions_size() - 1, in InferFftShape()
1953 result.set_dimensions(result.dimensions_size() - 1, in InferFftShape()
2048 output_shape.set_dimensions( in InferAllGatherShape()
2115 output_shape.set_dimensions(scatter_dimension, in InferReduceScatterShape()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc147 partition_shape.set_dimensions(i, tile_limit[i] - tile_offset[i]); in MakeNonPaddedShapeForGivenPartition()
149 partition_shape.set_dimensions(i, 0); in MakeNonPaddedShapeForGivenPartition()
212 padded_base_shape.set_dimensions( in GetPaddedShapeForUnevenPartitioning()
404 slice_shape.set_dimensions(dim, in TileToPartialReplicateHaloExchange()
515 slice_shape.set_dimensions(dim, dst_per_shard_size); in PadFromPartialReplicateShape()
534 padded_shape.set_dimensions(dim, result->shape().dimensions(dim) + in PadFromPartialReplicateShape()
766 halo_shape.set_dimensions(dim, halo_size); in ExchangeHalo()
791 self_shape.set_dimensions(dim, self_limit - self_start); in ExchangeHalo()
828 halo_shape.set_dimensions(dim, halo_size); in ExchangeHalo()
853 concat_shape.set_dimensions(dim, concat_dim_size); in ExchangeHalo()
[all …]
Dcanonicalize_all_gather_for_cse.cc75 new_ag_shape.set_dimensions( in RunOnComputation()
Dspmd_partitioner.cc716 shard_shape.set_dimensions( in ReshardAsWindowedInput()
751 halo_exchange_base_shape.set_dimensions(i, in ReshardAsWindowedInput()
762 halo_exchange_base_shape.set_dimensions( in ReshardAsWindowedInput()
795 halo_exchange_base_shape.set_dimensions(i, in ReshardAsWindowedInput()
829 halo_exchange_base_shape.set_dimensions( in ReshardAsWindowedInput()
865 shard_shape.set_dimensions(i, max_shard_size); in ReshardAsWindowedInput()
866 padded_shape.set_dimensions( in ReshardAsWindowedInput()
979 pad_hlo_shape.set_dimensions(i, hlo_->shape().dimensions(i)); in ReshardAsWindowedInput()
1196 target_shape.set_dimensions(i, base_shape().dimensions(i)); in ReplicatePartial()
1202 padded_target_shape.set_dimensions( in ReplicatePartial()
[all …]
Dcustom_call_handler.cc288 halo_shape.set_dimensions(dim, halo_size); in HandleCustomCallSPMDInternal_RotateRight()
291 halo_shape.set_dimensions(dim, halo_size); in HandleCustomCallSPMDInternal_RotateRight()
Dfft_handler.cc94 slice_shape.set_dimensions(concat->shape().rank() - 1, in PadEachPartitionWithHaloExchange()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.cc57 grouped_filter_shape.set_dimensions(input_feature_dim, 1); in GroupedFilterShapeForDepthwiseConvolution()
58 grouped_filter_shape.set_dimensions(output_feature_dim, in GroupedFilterShapeForDepthwiseConvolution()
72 new_shape.set_dimensions(num_dims - 1, num_groups); in TransposeFilterForGroupConvolutionBackpropInput()
/external/crosvm/devices/src/virtio/video/
Dffmpeg.rs98 builder.set_dimensions(Dimensions { in try_as_av_frame()
/external/crosvm/media/ffmpeg/src/
Davcodec.rs196 pub fn set_dimensions(&mut self, dimensions: Dimensions) { in set_dimensions() method
836 pub fn set_dimensions(&mut self, dimensions: Dimensions) -> Result<(), AvFrameError> { in set_dimensions() method
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Ddynamic_shaped_ops.cc61 result.set_dimensions(dim, shape->dimensions(dim)); in FindMaxShape()
/external/tensorflow/tensorflow/compiler/xla/
Dshape.h122 void set_dimensions(int index, int64_t value) { in set_dimensions() function
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Ddnn.h292 tensor_.set_dimensions(0, value); in set_count()
296 tensor_.set_dimensions(1, value); in set_feature_map_count()
412 tensor_.set_dimensions(0, value); in set_output_feature_map_count()
416 tensor_.set_dimensions(1, value); in set_input_feature_map_count()
/external/crosvm/devices/src/virtio/video/encoder/backend/
Dffmpeg.rs483 b.set_dimensions(Dimensions { in start_session()

12