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.cc102 auto padding_dimension = padding_config.mutable_dimensions(dim); in MakeSpatialPaddingConfig() local
103 padding_dimension->set_edge_padding_low(spatial_padding[i].first); in MakeSpatialPaddingConfig()
104 padding_dimension->set_edge_padding_high(spatial_padding[i].second); in MakeSpatialPaddingConfig()
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc3356 for (auto& padding_dimension : pad->padding_config().dimensions()) { in HandlePad() local
3357 if (padding_dimension.edge_padding_low() < 0 || in HandlePad()
3358 padding_dimension.edge_padding_high() < 0) { in HandlePad()
3361 if (padding_dimension.edge_padding_low() != 0 || in HandlePad()
3362 padding_dimension.edge_padding_high() != 0) { in HandlePad()
3365 } else if (padding_dimension.interior_padding()) { in HandlePad()
3465 PaddingConfig::PaddingConfigDimension* padding_dimension = in HandlePad() local
3468 if (padding_dimension->edge_padding_low() < 0) { in HandlePad()
3469 padding_dimension->set_edge_padding_low(0); in HandlePad()
3471 if (padding_dimension->edge_padding_high() < 0) { in HandlePad()
[all …]
Dalgebraic_simplifier_test.cc3299 for (auto& padding_dimension : pad->padding_config().dimensions()) { in TEST_F() local
3300 if (padding_dimension.edge_padding_low() < 0 || in TEST_F()
3301 padding_dimension.edge_padding_high() < 0) { in TEST_F()
3351 for (auto& padding_dimension : pad->padding_config().dimensions()) { in TEST_F() local
3352 if (padding_dimension.edge_padding_low() < 0 || in TEST_F()
3353 padding_dimension.edge_padding_high() < 0) { in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2007 for (auto& padding_dimension : pad->padding_config().dimensions()) { in HandlePad() local
2008 if (padding_dimension.edge_padding_low() < 0 || in HandlePad()
2009 padding_dimension.edge_padding_high() < 0) { in HandlePad()