Package apple.metalperformanceshadersgraph.enums

Class MPSGraphPoolingReturnIndicesMode

  • public final class MPSGraphPoolingReturnIndicesMode
extends java.lang.Object
    The flattening mode for returned indices with max pooling.

    • Field Summary

      Fields 
      Modifier and Type Field Description
      static long GlobalFlatten1D
      Returns indices flattened in inner most (last) dimension.
      static long GlobalFlatten2D
      Returns indices flattened in 2 innermost dimensions.
      static long GlobalFlatten3D
      Returns indices flattened in 3 innernost dimensions.
      static long GlobalFlatten4D
      Returns indices flattened in 4 innermost dimensions.
      static long LocalFlatten1D
      Returns indices within pooling window, flattened in inner most dimension.
      static long LocalFlatten2D
      Returns indices within pooling window, flattened in 2 innermost dimensions.
      static long LocalFlatten3D
      Returns indices within pooling window, flattened in 3 innernost dimensions.
      static long LocalFlatten4D
      Returns indices within pooling window, flattened in 4 innermost dimensions.
      static long None
      No indices returned.

    • Method Summary

      • Methods inherited from class java.lang.Object

        clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

    • Field Detail

      • GlobalFlatten1D

        public static final long GlobalFlatten1D
        Returns indices flattened in inner most (last) dimension.
        See Also:
        Constant Field Values

      • GlobalFlatten2D

        public static final long GlobalFlatten2D
        Returns indices flattened in 2 innermost dimensions. eg: HW in NCHW.
        See Also:
        Constant Field Values

      • GlobalFlatten3D

        public static final long GlobalFlatten3D
        Returns indices flattened in 3 innernost dimensions. eg: HWC in NHWC.
        See Also:
        Constant Field Values

      • GlobalFlatten4D

        public static final long GlobalFlatten4D
        Returns indices flattened in 4 innermost dimensions.
        See Also:
        Constant Field Values

      • LocalFlatten1D

        public static final long LocalFlatten1D
        Returns indices within pooling window, flattened in inner most dimension.
        See Also:
        Constant Field Values

      • LocalFlatten2D

        public static final long LocalFlatten2D
        Returns indices within pooling window, flattened in 2 innermost dimensions. eg: HW in NCHW.
        See Also:
        Constant Field Values

      • LocalFlatten3D

        public static final long LocalFlatten3D
        Returns indices within pooling window, flattened in 3 innernost dimensions. eg: HWC in NHWC.
        See Also:
        Constant Field Values

      • LocalFlatten4D

        public static final long LocalFlatten4D
        Returns indices within pooling window, flattened in 4 innermost dimensions.
        See Also:
        Constant Field Values