Searched defs:window_dim (Results 1 – 14 of 14) sorted by relevance
919 WindowDimension* window_dim = input_window->mutable_dimensions(dim_index); in RewriteInputWithDynamicPadding() local992 const WindowDimension& window_dim = window.dimensions(spatial_dim_index); in RewriteDynamicConvolutionInputGrad() local1056 const WindowDimension& window_dim = window.dimensions(spatial_dim_index); in RewriteDynamicConvolutionForward() local1134 const WindowDimension& window_dim = window.dimensions(spatial_dim_index); in RewriteDynamicConvolutionKernelGrad() local1193 const WindowDimension& window_dim = window.dimensions(dim_index); in RewriteDynamicReduceWindowSamePadding() local1230 const WindowDimension& window_dim = window.dimensions(dim_index); in RewriteDynamicSelectAndScatterSamePadding() local1277 const WindowDimension& window_dim = window.dimensions(dim_index); in RewriteDynamicSelectAndScatterSamePadding() local
298 auto window_dim = window.add_dimensions(); in HandleBatchGroupCount() local653 auto window_dim = window.add_dimensions(); in HandleConvolution() local
761 WindowDimension window_dim = hlo->window().dimensions(spatial_dim_index); in HandleDynamicConvolutionForward() local779 const WindowDimension& window_dim = window.dimensions(dimension); in HandleDynamicWindowSamePadding() local1237 const WindowDimension& window_dim = in HandleReduceWindow() local
116 for (const auto& window_dim : window_dims) { in AbslHashValue() local
109 for (auto window_dim : update_window_dims) { in PermuteScatterAndWindowDims() local
642 const auto& window_dim = window.dimensions(spatial_dimension); in GetConvolutionFlops() local
3579 for (int64_t window_dim : dim_numbers.update_window_dims()) { in ValidateScatterDimensionNumbers() local
5718 for (const WindowDimension& window_dim : window.dimensions()) { in HandleReduceWindow() local5915 auto& window_dim = *new_window.mutable_dimensions(i); in HandleReduceWindow() local
3403 auto window_dim = new_window.add_dimensions(); in PropagateOnBackpropFilterConv() local
3085 const WindowDimension& window_dim) { in EmitConvolution()
44 WindowDimension* window_dim = default_conv_window_.add_dimensions(); in GpuConvRewriterTest() local316 const WindowDimension& window_dim = custom_call->window().dimensions(i); in TEST_F() local455 const WindowDimension& window_dim = custom_call->window().dimensions(i); in TEST_F() local
196 void IncreasePaddingLowBy(int64_t delta, WindowDimension* window_dim) { in IncreasePaddingLowBy()200 void IncreasePaddingHighBy(int64_t delta, WindowDimension* window_dim) { in IncreasePaddingHighBy()
111 for (const WindowDimension& window_dim : conv->window().dimensions()) { in MatchBackwardFilter() local327 for (const WindowDimension& window_dim : conv->window().dimensions()) { in MatchBackwardInput() local
306 const WindowDimension& window_dim = window.dimensions(i); in CreateNaiveMlirConv() local