Searched refs:padding_low (Results 1 - 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dpad_insertion.cc61 std::max<int64>(0LL, conv_window.dimensions(i).padding_low()));
95 std::max<int64>(0LL, -conv_window.dimensions(i).padding_low());
192 window_dim->set_padding_low(window_dim->padding_low() + delta);
211 // BackwardFilterConv(ABCD, xyz, padding_low=1, padding_high=2)
214 // BackwardFilterConv(ABCD0, xyz, padding_low=pading_high=1)
215 // We choose the lesser of padding_low and padding_high as the new padding.
224 int64 padding_low = backward_conv->window().dimensions(i).padding_low(); local
226 if (padding_low < 0 || padding_high < 0) {
233 int64 new_conv_padding = std::min(padding_low, padding_hig
293 int64 padding_low = backward_conv->window().dimensions(i).padding_low(); local
353 int64 padding_low = backward_conv->window().dimensions(i).padding_low(); local
[all...]
H A Dcudnn_convolution_rewriter.cc104 if (window_dim.padding_low() < 0) {
136 dim->set_padding_low(conv->window().dimensions(i).padding_low());
145 // input_size + dim->padding_low() + min_padding_high
153 padded_input_size - input_size - dim->padding_low();
155 CHECK_GE(dim->padding_low(), 0);
159 // that range, we pick the one that's closest to dim->padding_low() and let
163 if (dim->padding_low() >= min_padding_high &&
164 dim->padding_low() <= max_padding_high) {
165 dim->set_padding_high(dim->padding_low());
167 if (dim->padding_low() < min_padding_hig
[all...]
H A Dcudnn_convolution_runner.cc113 CHECK_EQ(dim.padding_low(), dim.padding_high());
147 window.dimensions(dim).padding_low())
H A Dcudnn_convolution_rewriter_test.cc223 // Uneven padding: padding_low=0, padding_high=1
294 EXPECT_EQ(3, window_dim.padding_low());
421 EXPECT_EQ(0, window_dim.padding_low());
476 // padding_low=2, padding_high=1, base_dilation=2)
519 EXPECT_EQ(0, backward_conv_col_dim.padding_low());
527 // [3] = conv([4], [2], padding_low=1, padding_high=-1)
H A Delemental_ir_emitter.cc385 ir_builder_->getInt64(window.dimensions(i).padding_low()));
H A Dir_emitter_unnested.cc1819 ir_builder_.getInt64(window.dimensions(i).padding_low()));
/external/tensorflow/tensorflow/compiler/xla/
H A Dwindow_util.cc58 if (dim.padding_low() != 0) {
59 StrAppend(&str, ",padding_low=", dim.padding_low());
101 return StrCat(dim.padding_low(), "_", dim.padding_high());
133 if (dim.padding_low() != 0 || dim.padding_high() != 0) {
143 return dim.padding_low() == dim.padding_high();
158 return dim.padding_low() < 0 || dim.padding_high() < 0;
196 window_dim.padding_low() == 0 && window_dim.padding_high() == 0;
H A Dxla_data.proto363 int64 padding_low = 3;
365 // As padding_low, but on the high end of this dimension. For
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emitter.cc577 ir_builder_.getInt64(window.dimensions(i).padding_low()));
708 ir_builder_.getInt64(window.dimensions(i).padding_low()));
917 int64 padding_top = window.dimensions(0).padding_low();
920 one_dim_convolution ? 0 : window.dimensions(1).padding_low();
1053 ir_builder_.getInt64(window_dim.padding_low()));
/external/tensorflow/tensorflow/compiler/xla/service/
H A Dalgebraic_simplifier_test.cc2298 EXPECT_EQ(root->window().dimensions(0).padding_low(), 10);
2299 EXPECT_EQ(root->window().dimensions(1).padding_low(), 11);
2300 EXPECT_EQ(root->window().dimensions(2).padding_low(), 10);
2301 EXPECT_EQ(root->window().dimensions(3).padding_low(), 10);
H A Dshape_inference.cc268 if (!allow_negative_padding && dim.padding_low() < 0) {
290 dim.padding_low() + dilated_base + dim.padding_high();
H A Dhlo_evaluator.cc185 window_index[i] - window.dimensions(i).padding_low();
978 window_dim.padding_low() +
H A Dalgebraic_simplifier.cc1813 window_dim.set_padding_low(window_dim.padding_low() +

Completed in 219 milliseconds