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 delta, WindowDimension* window_dim) { in IncreasePaddingLowBy()197 void IncreasePaddingHighBy(int64 delta, WindowDimension* window_dim) { in IncreasePaddingHighBy()
190 for (const WindowDimension& window_dim : conv->window().dimensions()) { in MatchBackwardFilter() local379 for (const WindowDimension& window_dim : conv->window().dimensions()) { in MatchBackwardInput() local
766 WindowDimension* window_dim = input_window->mutable_dimensions(dim_index); in RewriteInputWithDynamicPadding() local835 const WindowDimension& window_dim = window.dimensions(spatial_dim_index); in RewriteDynamicConvolutionInputGrad() local899 const WindowDimension& window_dim = window.dimensions(spatial_dim_index); in RewriteDynamicConvolutionForward() local977 const WindowDimension& window_dim = window.dimensions(spatial_dim_index); in RewriteDynamicConvolutionKernelGrad() local1038 const WindowDimension& window_dim = window.dimensions(dim_index); in RewriteDynamicReduceWindowSamePadding() local1077 const WindowDimension& window_dim = window.dimensions(dim_index); in RewriteDynamicSelectAndScatterSamePadding() local1126 const WindowDimension& window_dim = window.dimensions(dim_index); in RewriteDynamicSelectAndScatterSamePadding() local
293 auto window_dim = window.add_dimensions(); in HandleBatchGroupCount() local646 auto window_dim = window.add_dimensions(); in HandleConvolution() local
106 for (auto window_dim : update_window_dims) { in PermuteScatterAndWindowDims() local
713 WindowDimension window_dim = hlo->window().dimensions(spatial_dim_index); in HandleDynamicConvolutionForward() local731 const WindowDimension& window_dim = window.dimensions(dimension); in HandleDynamicWindowSamePadding() local1064 const WindowDimension& window_dim = in HandleReduceWindow() local
624 const auto& window_dim = window.dimensions(spatial_dimension); in HandleConvolution() local
3460 for (int64 window_dim : dim_numbers.update_window_dims()) { in ValidateScatterDimensionNumbers() local
2520 auto window_dim = new_window.add_dimensions(); in PropagateOnBackpropFilterConv() local
2874 const WindowDimension& window_dim) { in EmitConvolution()
4975 auto& window_dim = *new_window.mutable_dimensions(i); in HandleReduceWindow() local
400 for (const auto& window_dim : reduce_window_op.window_dimensions()) { in CreateParallelLoopsToTraverseOutputAndWindow() local556 for (const auto& window_dim : in InsertWindowLoops() local
304 const WindowDimension& window_dim = window.dimensions(i); in CreateNaiveMlirConv() local