Class DnnOp

    • Constructor Detail

      • DnnOp

        public DnnOp​(String l,
                     Types.DataType dt,
                     Types.ValueType vt,
                     Types.OpOpDnn o,
                     ArrayList<Hop> inp)
        Create a hop from the builtin expression
        Parameters:
        l - name of the hop
        dt - datatype (only supports matrix datatype)
        vt - valuetype (only supports matrix valuetype)
        o - type of this hop
        inp - input hops
    • Method Detail

      • isGPUEnabled

        public boolean isGPUEnabled()
        Description copied from class: Hop
        In memory-based optimizer mode (see OptimizerUtils.isMemoryBasedOptLevel()), the exectype is determined by checking this method as well as memory budget of this Hop. Please see findExecTypeByMemEstimate for more detail. This method is necessary because not all operator are supported efficiently on GPU (for example: operations on frames and scalar as well as operations such as table).
        Specified by:
        isGPUEnabled in class Hop
        Returns:
        true if the Hop is eligible for GPU Exectype.
      • isStride1Pad0

        public boolean isStride1Pad0()
      • allowsAllExecTypes

        public boolean allowsAllExecTypes()
        Specified by:
        allowsAllExecTypes in class Hop
      • refreshSizeInformation

        public void refreshSizeInformation()
        Description copied from class: Hop
        Update the output size information for this hop.
        Specified by:
        refreshSizeInformation in class Hop
      • compare

        public boolean compare​(Hop that)
        Specified by:
        compare in class Hop