// by SAME or VALID tensorflow padding.
def HasSameOrValidPadding: Constraint<Neg<HasCustomPadding.predicate>>;

// See the function doc in the header file.
def GetPadOpAttr: NativeCodeCall<"GetPadOpAttr($_builder, (*$0.begin()).getDefiningOp<mhlo::CompositeOp>())">;

// See the function doc in the header file.

namespace odml {

// Given a Composite op that wraps a core.aten.avg_pool2d, returns the padding
// configuration required for the `tfl.pad` if the padding part of the op is
// to be done before average pooling.
DenseIntElementsAttr GetPadOpAttr(Builder& builder, mhlo::CompositeOp op);

// Given a Composite op that wraps a core.aten.avg_pool2d, and assuming that
// the padding part is extracted into a tfl.pad op prior to a

    const int ceil_pad_right = pool.kw - remaining_right;
    pad_right = ceil_pad_right - pool.pw;
  }

  return {pool.ph, pad_bottom, pool.pw, pad_right};
}

DenseIntElementsAttr GetPadOpAttr(Builder& builder, CompositeOp op) {
  const TorchAvgPoolData pool = GetTorchAvgPoolData(op);

  const auto values = GetPadOpPaddingValues(pool);

  llvm::SmallVector<int32_t> padding_vec(8, 0);  // NHWC