Searched defs:padding_config_dim (Results 1 – 4 of 4) sorted by relevance
542 PaddingConfig::PaddingConfigDimension padding_config_dim; in PadVectorWithZeros() local
1380 auto padding_config_dim = padding_config.add_dimensions(); in HandleConcatenate() local4676 auto padding_config_dim = padding_config.add_dimensions(); in HandleDynamicUpdateSlice() local
258 auto padding_config_dim = padding_config.add_dimensions(); in PadToShape() local1015 auto padding_config_dim = padding_config.add_dimensions(); in ExchangeHaloAndGetValidData() local
748 auto padding_config_dim = padding_config.add_dimensions(); in ReshardAsWindowedInput() local3963 auto* padding_config_dim = padding_config.mutable_dimensions(dim); in PreprocessHlos() local