avgPool function

VARP avgPool(
  1. VARP x,
  2. List<int> kernal, {
  3. PaddingMode pad = PaddingMode.VALID,
  4. List<int> stride = const [1, 1],
  5. List<int> pads = const [0, 0],
})

Implementation

VARP avgPool(
  VARP x,
  List<int> kernal, {
  PaddingMode pad = PaddingMode.VALID,
  List<int> stride = const [1, 1],
  List<int> pads = const [0, 0],
}) {
  final (kernalPtr, kernalSize) = kernal.toNativeArrayI32();
  final (stridePtr, strideSize) = stride.toNativeArrayI32();
  final (padPtr, padSize) = pads.toNativeArrayI32();
  final rval = VARP.fromPointer(
    C.mnn_expr_AvePool(
      x.ptr,
      kernalPtr.cast(),
      kernalSize,
      stridePtr.cast(),
      strideSize,
      pad.value,
      padPtr.cast(),
      padSize,
    ),
  );
  calloc.free(kernalPtr);
  calloc.free(stridePtr);
  calloc.free(padPtr);
  return rval;
}