Searched refs:window_reversal (Results 1 – 16 of 16) sorted by relevance
84 if (dim.window_reversal()) { in ToString()132 return StrCat(dim.window_reversal() ? 1 : 0); in ToString()196 if (dim.window_reversal()) { in HasWindowReversal()207 bool reversed = window.dimensions()[0].window_reversal(); in AllOrNoneReversed()209 return dim.window_reversal() == reversed; in AllOrNoneReversed()
457 bool window_reversal = 7; field
37 !wd.window_reversal()) { in ConvSpatialDimensionIsParallel()43 wd.padding_low() == lhs_size - 1 && wd.window_reversal() && in ConvSpatialDimensionIsParallel()78 wd.padding_low() == 0 && !wd.window_reversal()) { in ParseConvolutionDimsInfo()86 wd.padding_low() == 0 && !wd.window_reversal()) { in ParseConvolutionDimsInfo()92 wd.padding_low() == rhs_size - 1 && wd.window_reversal()) { in ParseConvolutionDimsInfo()
1462 consumer->window().dimensions(dim).window_reversal()); in Propagate()
5298 !window_dims[spatial_dim].window_reversal() && in SwapConvOperands()5330 if (!window_dims[spatial_dim].window_reversal()) { in SwapConvOperands()
1215 (window_dim.window_reversal()
2939 window.dimensions(i).window_reversal() in EmitConvolution()
7147 EXPECT_EQ(conv->window().dimensions(0).window_reversal(), true); in TEST_F()7148 EXPECT_EQ(conv->window().dimensions(1).window_reversal(), true); in TEST_F()
342 window.dimensions_size() > 0 && window.dimensions()[0].window_reversal(); in GetGpuConvConfig()351 CHECK_EQ(dims_reversed, dim.window_reversal()) << inst_as_string; in GetGpuConvConfig()
207 if (window_dim.window_reversal()) { in MatchBackwardFilter()392 if (window_dim.window_reversal()) { in MatchBackwardInput()
1161 mlir::DenseElementsAttr window_reversal = op.window_reversal().getValue(); in EmitConvolutionThunkFromMlir() local1175 dim->set_window_reversal(window_reversal.getValue<bool>(index)); in EmitConvolutionThunkFromMlir()
118 window_reversal = dense<true>: tensor<2xi1>}
298 // CHECK-SAME: window_reversal = dense<true> : tensor<2xi1>336 // CHECK-SAME: window_reversal = dense<false> : tensor<2xi1>
327 if (wd.base_dilation() != 1 || wd.window_reversal()) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()547 if (window.dimensions(i).window_reversal()) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
763 auto window_reversal = llvm::to_vector<4>(llvm::map_range( in EmitDnnConvolution() local765 [](const xla::WindowDimension& dim) { return dim.window_reversal(); })); in EmitDnnConvolution()768 op.window_reversalAttr(DenseElementsAttr::get(type, window_reversal)); in EmitDnnConvolution()
1011 OptionalAttr<BoolElementsAttr>:$window_reversal,