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

      • checkArity

        public void checkArity()
        Description copied from class: Hop
        Check whether this Hop has a correct number of inputs. (Some Hops can have a variable number of inputs, such as DataOp, DataGenOp, ParameterizedBuiltinOp, ReorgOp, TernaryOp, QuaternaryOp, MultipleOp, DnnOp, and SpoofFusedOp.) Parameterized Hops (such as DataOp) can check that the number of parameters matches the number of inputs.
        Specified by:
        checkArity in class Hop
      • 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