Searched refs:spatial_padding (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | pooling.cc | 30 absl::Span<const std::pair<int64, int64>> spatial_padding, in AvgPoolDivideByCountWithGeneralPadding() argument
35 const int num_spatial_dims = spatial_padding.size(); in AvgPoolDivideByCountWithGeneralPadding()
57 dims->set_edge_padding_low(spatial_padding[i].first); in AvgPoolDivideByCountWithGeneralPadding()
58 dims->set_edge_padding_high(spatial_padding[i].second); in AvgPoolDivideByCountWithGeneralPadding()
90 absl::Span<const std::pair<int64, int64>> spatial_padding, in MakeSpatialPaddingConfig() argument
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()
196 absl::Span<const std::pair<int64, int64>> spatial_padding, in AvgPoolGrad() argument
237 out_backprop, gradients_size, kernel_size, stride, spatial_padding, in AvgPoolGrad()
252 (spatial_padding[i].first + spatial_padding[i].second); in AvgPoolGrad()
[all …]
|
| D | pooling.h | 74 absl::Span<const std::pair<int64, int64>> spatial_padding,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | pooling_ops.cc | 239 auto spatial_padding = MakeSpatialPadding( in Compile() local
246 xla::AvgPool(converted_input, ksize, stride, spatial_padding, in Compile()
|