Searched defs:padding_dim (Results 1 – 6 of 6) sorted by relevance
167 auto* padding_dim = padding_config.add_dimensions(); in CompileInternal() local200 auto* padding_dim = slice_padding_config.add_dimensions(); in CompileInternal() local
158 for (const auto& padding_dim : padding_dims) { in AbslHashValue() local
908 PaddingConfig::PaddingConfigDimension padding_dim; in RewriteInputWithDynamicPadding() local920 auto* padding_dim = padding_configs.mutable_dimensions(shape_dim); in RewriteInputWithDynamicPadding() local1275 PaddingConfig::PaddingConfigDimension padding_dim; in RewriteDynamicSelectAndScatterSamePadding() local
5361 std::vector<int64_t> padding_dim; in ParsePaddingConfig() local
5719 auto& padding_dim = *padding_config.add_dimensions(); in HandleReduceWindow() local
1150 int64_t padding_dim = slice_sharding_dim; in PartitionBaseCase() local