Searched refs:y_in_bounds (Results 1 – 2 of 2) sorted by relevance
115 auto y_in_bounds = [input_height = ctx.input_shapes[0][1], in GenerateAveragePoolingCode() local129 y_in_bounds(base_y) && in GenerateAveragePoolingCode()130 y_in_bounds(base_y + output_shape_max_y * attr.strides.h)); in GenerateAveragePoolingCode()
1602 Value y_in_bounds = rewriter.create<AndOp>( in matchAndRewrite() local1611 rewriter.create<AndOp>(loc, x_in_bounds, y_in_bounds)); in matchAndRewrite()