Searched defs:window_dim (Results 1 – 10 of 10) sorted by relevance
128 for (const WindowDimension& window_dim : conv->window().dimensions()) { in MatchBackwardFilter() local308 for (const WindowDimension& window_dim : conv->window().dimensions()) { in MatchBackwardInput() local
43 WindowDimension* window_dim = default_conv_window_.add_dimensions(); in CudnnConvRewriterTest() local312 const WindowDimension& window_dim = custom_call->window().dimensions(i); in TEST_F() local447 const WindowDimension& window_dim = custom_call->window().dimensions(i); in TEST_F() local
193 void IncreasePaddingLowBy(int64 delta, WindowDimension* window_dim) { in IncreasePaddingLowBy()197 void IncreasePaddingHighBy(int64 delta, WindowDimension* window_dim) { in IncreasePaddingHighBy()
106 for (auto window_dim : update_window_dims) { in PermuteScatterAndWindowDims() local
498 const auto& window_dim = window.dimensions(spatial_dimension); in HandleConvolution() local
3085 for (int64 window_dim : dim_numbers.update_window_dims()) { in ValidateScatterDimensionNumbers() local
1123 const auto& window_dim = window.dimensions(ki); in HandleConvolution() local
3292 auto& window_dim = *new_window.mutable_dimensions(i); in HandleReduceWindow() local
202 const WindowDimension& window_dim = window.dimensions(logical_dim); in IsInactiveWindowDimension() local
1003 const WindowDimension& window_dim) { in EmitElementalConvolution()