Searched refs:window_dim (Results 1 – 10 of 10) sorted by relevance
43 WindowDimension* window_dim = default_conv_window_.add_dimensions(); in CudnnConvRewriterTest() local44 window_dim->set_size(1); in CudnnConvRewriterTest()45 window_dim->set_stride(1); in CudnnConvRewriterTest()46 window_dim->set_padding_low(0); in CudnnConvRewriterTest()47 window_dim->set_padding_high(0); in CudnnConvRewriterTest()48 window_dim->set_window_dilation(1); in CudnnConvRewriterTest()49 window_dim->set_base_dilation(1); in CudnnConvRewriterTest()312 const WindowDimension& window_dim = custom_call->window().dimensions(i); in TEST_F() local315 EXPECT_EQ(3, window_dim.padding_low()); in TEST_F()316 EXPECT_EQ(3, window_dim.padding_high()); in TEST_F()[all …]
128 for (const WindowDimension& window_dim : conv->window().dimensions()) { in MatchBackwardFilter() local129 if (window_dim.stride() != 1) { in MatchBackwardFilter()135 if (window_dim.base_dilation() != 1) { in MatchBackwardFilter()141 if (window_dim.padding_low() < 0) { in MatchBackwardFilter()145 if (window_dim.window_reversal()) { in MatchBackwardFilter()308 for (const WindowDimension& window_dim : conv->window().dimensions()) { in MatchBackwardInput() local309 if (window_dim.stride() != 1) { in MatchBackwardInput()315 if (window_dim.window_dilation() != 1) { in MatchBackwardInput()321 if (window_dim.window_reversal()) { in MatchBackwardInput()
193 void IncreasePaddingLowBy(int64 delta, WindowDimension* window_dim) { in IncreasePaddingLowBy() argument194 window_dim->set_padding_low(window_dim->padding_low() + delta); in IncreasePaddingLowBy()197 void IncreasePaddingHighBy(int64 delta, WindowDimension* window_dim) { in IncreasePaddingHighBy() argument198 window_dim->set_padding_high(window_dim->padding_high() + delta); in IncreasePaddingHighBy()
202 const WindowDimension& window_dim = window.dimensions(logical_dim); in IsInactiveWindowDimension() local203 return window_dim.size() == 1 && window_dim.stride() == 1 && in IsInactiveWindowDimension()204 window_dim.padding_low() == 0 && window_dim.padding_high() == 0; in IsInactiveWindowDimension()
498 const auto& window_dim = window.dimensions(spatial_dimension); in HandleConvolution() local499 const int64 undilated_index = output_idx * window_dim.stride() - in HandleConvolution()500 window_dim.padding_low() + in HandleConvolution()501 kernel_idx * window_dim.window_dilation(); in HandleConvolution()506 window_dim.base_dilation() > 1 in HandleConvolution()507 ? undilated_index / window_dim.base_dilation() in HandleConvolution()511 if (undilated_index != lhs_spatial_index * window_dim.base_dilation()) { in HandleConvolution()
106 for (auto window_dim : update_window_dims) { in PermuteScatterAndWindowDims() local107 permutation.push_back(window_dim); in PermuteScatterAndWindowDims()
1123 const auto& window_dim = window.dimensions(ki); in HandleConvolution() local1125 out_index[output_spatial_dim] * window_dim.stride() - in HandleConvolution()1126 window_dim.padding_low() + in HandleConvolution()1127 rhs_spatial_index[ki] * window_dim.window_dilation(); in HandleConvolution()1130 if (window_dim.base_dilation() > 1 && in HandleConvolution()1131 undilated_index % window_dim.base_dilation() != 0) { in HandleConvolution()1138 if (window_dim.base_dilation() > 1) { in HandleConvolution()1139 lhs_spatial_index = undilated_index / window_dim.base_dilation(); in HandleConvolution()1153 (window_dim.window_reversal() in HandleConvolution()1154 ? ((window_dim.size() - 1) - rhs_spatial_index[ki]) in HandleConvolution()
3085 for (int64 window_dim : dim_numbers.update_window_dims()) { in ValidateScatterDimensionNumbers() local3086 if (window_dim < 0 || window_dim >= updates_rank) { in ValidateScatterDimensionNumbers()3090 updates_rank, window_dim); in ValidateScatterDimensionNumbers()
3292 auto& window_dim = *new_window.mutable_dimensions(i); in HandleReduceWindow() local3293 window_dim.set_padding_low(window_dim.padding_low() + in HandleReduceWindow()3295 window_dim.set_padding_high(window_dim.padding_high() + in HandleReduceWindow()
1003 const WindowDimension& window_dim) { in EmitElementalConvolution() argument1005 NSWMul(output_index, b_.getInt64(window_dim.stride())); in EmitElementalConvolution()1007 NSWMul(kernel_index, b_.getInt64(window_dim.window_dilation())); in EmitElementalConvolution()1009 b_.getInt64(window_dim.padding_low())); in EmitElementalConvolution()