[(IsBoolCompositeAttribute<"ceil_mode", "false"> $attrs),
                    (IsBoolCompositeAttribute<"count_include_pad", "false"> $attrs),
                    (IsStrCompositeAttribute<"divisor_override", "py_None"> $attrs),
                    (HasSameOrValidPadding $old_val)]>;

// Replaces an aten.avg_pool2d with (T -> tfl.pad -> tfl.average_pool_2d  -> T).
def LegalizeAvgPool2dWithPadComposite: Pat<

// CHECK:         }

func.func @avg_pool2d_1(%arg0: tensor<1x3x6x6xf32>) -> (tensor<*xf32>) {
  %0 = mhlo.composite "aten.avg_pool2d.default" %arg0 {composite_attributes = {ceil_mode = false, count_include_pad = true, divisor_override = "py_None", kernel_size = dense<3> : tensor<2xi64>, padding = dense<0> : tensor<2xi64>, stride = dense<1> : tensor<2xi64>}, decomposition = @XlaCallModule_aten.avg_pool2d.default.impl_0} : (tensor<1x3x6x6xf32>) -> tensor<1x3x4x4xf32>