Searched defs:window_dim (Results 1 – 14 of 14) sorted by relevance
43 WindowDimension* window_dim = default_conv_window_.add_dimensions(); in GpuConvRewriterTest() local315 const WindowDimension& window_dim = custom_call->window().dimensions(i); in TEST_F() local454 const WindowDimension& window_dim = custom_call->window().dimensions(i); in TEST_F() local
193 void IncreasePaddingLowBy(int64_t delta, WindowDimension* window_dim) { in IncreasePaddingLowBy()197 void IncreasePaddingHighBy(int64_t delta, WindowDimension* window_dim) { in IncreasePaddingHighBy()
190 for (const WindowDimension& window_dim : conv->window().dimensions()) { in MatchBackwardFilter() local406 for (const WindowDimension& window_dim : conv->window().dimensions()) { in MatchBackwardInput() local
864 WindowDimension* window_dim = input_window->mutable_dimensions(dim_index); in RewriteInputWithDynamicPadding() local934 const WindowDimension& window_dim = window.dimensions(spatial_dim_index); in RewriteDynamicConvolutionInputGrad() local999 const WindowDimension& window_dim = window.dimensions(spatial_dim_index); in RewriteDynamicConvolutionForward() local1078 const WindowDimension& window_dim = window.dimensions(spatial_dim_index); in RewriteDynamicConvolutionKernelGrad() local1139 const WindowDimension& window_dim = window.dimensions(dim_index); in RewriteDynamicReduceWindowSamePadding() local1178 const WindowDimension& window_dim = window.dimensions(dim_index); in RewriteDynamicSelectAndScatterSamePadding() local1227 const WindowDimension& window_dim = window.dimensions(dim_index); in RewriteDynamicSelectAndScatterSamePadding() local
293 auto window_dim = window.add_dimensions(); in HandleBatchGroupCount() local648 auto window_dim = window.add_dimensions(); in HandleConvolution() local
106 for (auto window_dim : update_window_dims) { in PermuteScatterAndWindowDims() local
732 WindowDimension window_dim = hlo->window().dimensions(spatial_dim_index); in HandleDynamicConvolutionForward() local750 const WindowDimension& window_dim = window.dimensions(dimension); in HandleDynamicWindowSamePadding() local1099 const WindowDimension& window_dim = in HandleReduceWindow() local
625 const auto& window_dim = window.dimensions(spatial_dimension); in HandleConvolution() local
3597 for (int64_t window_dim : dim_numbers.update_window_dims()) { in ValidateScatterDimensionNumbers() local
2912 auto window_dim = new_window.add_dimensions(); in PropagateOnBackpropFilterConv() local
3059 const WindowDimension& window_dim) { in EmitConvolution()
5294 auto& window_dim = *new_window.mutable_dimensions(i); in HandleReduceWindow() local
406 for (const auto& window_dim : reduce_window_op.window_dimensions()) { in CreateParallelLoopsToTraverseOutputAndWindow() local563 for (const auto& window_dim : in InsertWindowLoops() local
305 const WindowDimension& window_dim = window.dimensions(i); in CreateNaiveMlirConv() local