Searched refs:window_dim (Results 1 – 7 of 7) sorted by relevance
40 WindowDimension* window_dim = default_conv_window_.add_dimensions(); in CudnnConvolutionRewriterTest() local41 window_dim->set_size(1); in CudnnConvolutionRewriterTest()42 window_dim->set_stride(1); in CudnnConvolutionRewriterTest()43 window_dim->set_padding_low(0); in CudnnConvolutionRewriterTest()44 window_dim->set_padding_high(0); in CudnnConvolutionRewriterTest()45 window_dim->set_window_dilation(1); in CudnnConvolutionRewriterTest()46 window_dim->set_base_dilation(1); in CudnnConvolutionRewriterTest()291 const WindowDimension& window_dim = custom_call->window().dimensions(i); in TEST_F() local294 EXPECT_EQ(3, window_dim.padding_low()); in TEST_F()295 EXPECT_EQ(3, window_dim.padding_high()); in TEST_F()[all …]
91 for (const WindowDimension& window_dim : conv->window().dimensions()) { in MatchBackwardFilter() local92 if (window_dim.stride() != 1) { in MatchBackwardFilter()98 if (window_dim.base_dilation() != 1) { in MatchBackwardFilter()104 if (window_dim.padding_low() < 0) { in MatchBackwardFilter()108 if (window_dim.window_reversal()) { in MatchBackwardFilter()255 for (const WindowDimension& window_dim : conv->window().dimensions()) { in MatchBackwardInput() local256 if (window_dim.stride() != 1) { in MatchBackwardInput()262 if (window_dim.window_dilation() != 1) { in MatchBackwardInput()268 if (window_dim.window_reversal()) { in MatchBackwardInput()
191 void IncreasePaddingLowBy(int64 delta, WindowDimension* window_dim) { in IncreasePaddingLowBy() argument192 window_dim->set_padding_low(window_dim->padding_low() + delta); in IncreasePaddingLowBy()195 void IncreasePaddingHighBy(int64 delta, WindowDimension* window_dim) { in IncreasePaddingHighBy() argument196 window_dim->set_padding_high(window_dim->padding_high() + delta); in IncreasePaddingHighBy()
194 const WindowDimension& window_dim = window.dimensions(logical_dim); in IsInactiveWindowDimension() local195 return window_dim.size() == 1 && window_dim.stride() == 1 && in IsInactiveWindowDimension()196 window_dim.padding_low() == 0 && window_dim.padding_high() == 0; in IsInactiveWindowDimension()
975 const auto& window_dim = window.dimensions(ki); in HandleConvolution() local977 out_index[output_spatial_dim] * window_dim.stride() - in HandleConvolution()978 window_dim.padding_low() + in HandleConvolution()979 rhs_spatial_index[ki] * window_dim.window_dilation(); in HandleConvolution()982 if (window_dim.base_dilation() > 1 && in HandleConvolution()983 undilated_index % window_dim.base_dilation() != 0) { in HandleConvolution()989 if (window_dim.base_dilation() > 1) { in HandleConvolution()991 undilated_index / window_dim.base_dilation(); in HandleConvolution()1004 window_dim.window_reversal() in HandleConvolution()1005 ? ((window_dim.size() - 1) - rhs_spatial_index[ki]) in HandleConvolution()
1812 auto& window_dim = *new_window.mutable_dimensions(i); in HandleReduceWindow() local1813 window_dim.set_padding_low(window_dim.padding_low() + in HandleReduceWindow()1815 window_dim.set_padding_high(window_dim.padding_high() + in HandleReduceWindow()
1045 const WindowDimension& window_dim) { in HandleConvolution() argument1047 output_index, ir_builder_.getInt64(window_dim.stride())); in HandleConvolution()1050 ir_builder_.getInt64(window_dim.window_dilation())); in HandleConvolution()1053 ir_builder_.getInt64(window_dim.padding_low())); in HandleConvolution()