Searched defs:window_dim (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dcudnn_convolution_rewriter_test.cc40 WindowDimension* window_dim = default_conv_window_.add_dimensions(); local
41 window_dim->set_size(1);
42 window_dim->set_stride(1);
43 window_dim->set_padding_low(0);
44 window_dim->set_padding_high(0);
45 window_dim->set_window_dilation(1);
46 window_dim->set_base_dilation(1);
291 const WindowDimension& window_dim = custom_call->window().dimensions(i); local
294 EXPECT_EQ(3, window_dim.padding_low());
295 EXPECT_EQ(3, window_dim
420 const WindowDimension& window_dim = custom_call->window().dimensions(i); local
[all...]
H A Dpad_insertion.cc191 void IncreasePaddingLowBy(int64 delta, WindowDimension* window_dim) { argument
192 window_dim->set_padding_low(window_dim->padding_low() + delta);
195 void IncreasePaddingHighBy(int64 delta, WindowDimension* window_dim) { argument
196 window_dim->set_padding_high(window_dim->padding_high() + delta);
/external/tensorflow/tensorflow/compiler/xla/service/
H A Dalgebraic_simplifier.cc1812 auto& window_dim = *new_window.mutable_dimensions(i); local
1813 window_dim.set_padding_low(window_dim.padding_low() +
1815 window_dim.set_padding_high(window_dim.padding_high() +

Completed in 103 milliseconds