[(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<

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>

                    "Failed to convert Toco to Python String.");
    return nullptr;
  }

  tensorflow::GraphDebugInfo debug_info;
  if (debug_info_txt_raw && debug_info_txt_raw != Py_None) {
    std::string debug_info_txt = ConvertArg(debug_info_txt_raw, &error);
    if (error) {
      PyErr_SetString(PyExc_ValueError, "Input DebugInfo is invalid.");
      return nullptr;
    }

Austin Clements <******@****.***> 1668185488 -0500