Home
last modified time | relevance | path

Searched defs:padding_config_dim (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_creation_utils.cc657 PaddingConfig::PaddingConfigDimension padding_config_dim; in PadVectorWithZeros() local
H A Dalgebraic_simplifier.cc1500 auto padding_config_dim = padding_config.add_dimensions(); in HandleConcatenate() local
5072 auto padding_config_dim = padding_config.add_dimensions(); in HandleDynamicUpdateSlice() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc970 auto padding_config_dim = padding_config.add_dimensions(); in ReshardAsWindowedInput() local
4770 auto* padding_config_dim = padding_config.mutable_dimensions(dim); in PreprocessHlos() local
H A Dspmd_partitioner_util.cc931 auto padding_config_dim = padding_config.add_dimensions(); in ExchangeHaloAndGetValidData() local