Searched refs:PaddingConfigDimension (Results 1 – 12 of 12) sorted by relevance
447 class PaddingConfigDimension(object): class482 dimension = PaddingConfigDimension()
374 xla::PaddingConfig::PaddingConfigDimension* config_dim =
1060 dimension = xla_client.PaddingConfigDimension()
164 [](const PaddingConfig::PaddingConfigDimension& dim) { in HasSymmetricPadding()
96 message PaddingConfigDimension { message109 repeated PaddingConfigDimension dimensions = 1;
538 PaddingConfig::PaddingConfigDimension padding_config_dim; in PadVectorWithZeros()
755 PaddingConfig::PaddingConfigDimension padding_dim; in RewriteInputWithDynamicPadding()1124 PaddingConfig::PaddingConfigDimension padding_dim; in RewriteDynamicSelectAndScatterSamePadding()
3681 [](const PaddingConfig::PaddingConfigDimension& dim) { in PaddingConfigToString()3686 [&](string* out, const PaddingConfig::PaddingConfigDimension& dim) { in PaddingConfigToString()
3465 PaddingConfig::PaddingConfigDimension* padding_dimension = in HandlePad()3491 const PaddingConfig::PaddingConfigDimension& padding_dimension = in HandlePad()
553 [](const PaddingConfig::PaddingConfigDimension& p) { in InferPadShape()
1842 PaddingConfig::PaddingConfigDimension* dimension = padding.add_dimensions(); in TEST_F()
1826 `PaddingConfig` is a repeated field of `PaddingConfigDimension`, which contains