Home
last modified time | relevance | path

Searched refs:padding_dimension (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling.cc101 auto padding_dimension = padding_config.mutable_dimensions(dim); in MakeSpatialPaddingConfig() local
102 padding_dimension->set_edge_padding_low(spatial_padding[i].first); in MakeSpatialPaddingConfig()
103 padding_dimension->set_edge_padding_high(spatial_padding[i].second); in MakeSpatialPaddingConfig()
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc2225 for (auto& padding_dimension : pad->padding_config().dimensions()) { in HandlePad() local
2226 if (padding_dimension.edge_padding_low() < 0 || in HandlePad()
2227 padding_dimension.edge_padding_high() < 0) { in HandlePad()
2230 if (padding_dimension.edge_padding_low() != 0 || in HandlePad()
2231 padding_dimension.edge_padding_high() != 0) { in HandlePad()
2253 PaddingConfig::PaddingConfigDimension* padding_dimension = in HandlePad() local
2256 if (padding_dimension->edge_padding_low() < 0) { in HandlePad()
2257 padding_dimension->set_edge_padding_low(0); in HandlePad()
2259 if (padding_dimension->edge_padding_high() < 0) { in HandlePad()
2260 padding_dimension->set_edge_padding_high(0); in HandlePad()
[all …]
Dalgebraic_simplifier_test.cc2587 for (auto& padding_dimension : pad->padding_config().dimensions()) { in TEST_F() local
2588 if (padding_dimension.edge_padding_low() < 0 || in TEST_F()
2589 padding_dimension.edge_padding_high() < 0) { in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2093 for (auto& padding_dimension : pad->padding_config().dimensions()) { in HandlePad() local
2094 if (padding_dimension.edge_padding_low() < 0 || in HandlePad()
2095 padding_dimension.edge_padding_high() < 0) { in HandlePad()