Home
last modified time | relevance | path

Searched refs:conv_window (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_rewriter_test.cc106 Window conv_window = default_conv_window_; in TEST_F() local
107 conv_window.mutable_dimensions(1)->set_size(2); in TEST_F()
108 conv_window.mutable_dimensions(1)->set_window_dilation(2); in TEST_F()
112 /*batch_group_count=*/1, conv_window, in TEST_F()
116 /*batch_group_count=*/1, conv_window, in TEST_F()
147 Window conv_window = default_conv_window_; in TEST_F() local
148 conv_window.mutable_dimensions(1)->set_size(3); in TEST_F()
152 /*batch_group_count=*/1, conv_window, in TEST_F()
156 /*batch_group_count=*/1, conv_window, in TEST_F()
178 Window conv_window = default_conv_window_; in TEST_F() local
[all …]
Dcudnn_conv_padding_legalization.cc49 Window* conv_window, const ConvolutionDimensionNumbers& conv_dnums, in MaybePaddedAndSlicedInput() argument
52 if (!window_util::HasSymmetricPadding(*conv_window) || in MaybePaddedAndSlicedInput()
53 window_util::HasBaseDilation(*conv_window)) { in MaybePaddedAndSlicedInput()
66 if (conv_window->dimensions(i).padding_low() > 0) { in MaybePaddedAndSlicedInput()
68 conv_window->dimensions(i).padding_low()); in MaybePaddedAndSlicedInput()
69 conv_window->mutable_dimensions(i)->set_padding_low(0); in MaybePaddedAndSlicedInput()
71 if (conv_window->dimensions(i).padding_high() > 0) { in MaybePaddedAndSlicedInput()
73 conv_window->dimensions(i).padding_high()); in MaybePaddedAndSlicedInput()
74 conv_window->mutable_dimensions(i)->set_padding_high(0); in MaybePaddedAndSlicedInput()
76 if (conv_window->dimensions(i).base_dilation() != 1) { in MaybePaddedAndSlicedInput()
[all …]