Searched refs:padding_low (Results 1 - 13 of 13) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | pad_insertion.cc | 61 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 D | cudnn_convolution_rewriter.cc | 104 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 D | cudnn_convolution_runner.cc | 113 CHECK_EQ(dim.padding_low(), dim.padding_high()); 147 window.dimensions(dim).padding_low())
|
H A D | cudnn_convolution_rewriter_test.cc | 223 // 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 D | elemental_ir_emitter.cc | 385 ir_builder_->getInt64(window.dimensions(i).padding_low()));
|
H A D | ir_emitter_unnested.cc | 1819 ir_builder_.getInt64(window.dimensions(i).padding_low()));
|
/external/tensorflow/tensorflow/compiler/xla/ |
H A D | window_util.cc | 58 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 D | xla_data.proto | 363 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 D | ir_emitter.cc | 577 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 D | algebraic_simplifier_test.cc | 2298 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 D | shape_inference.cc | 268 if (!allow_negative_padding && dim.padding_low() < 0) { 290 dim.padding_low() + dilated_base + dim.padding_high();
|
H A D | hlo_evaluator.cc | 185 window_index[i] - window.dimensions(i).padding_low(); 978 window_dim.padding_low() +
|
H A D | algebraic_simplifier.cc | 1813 window_dim.set_padding_low(window_dim.padding_low() +
|
Completed in 219 milliseconds