Class DnnOp

All Implemented Interfaces:
ParseInfo

public class DnnOp extends MultiThreadedHop
  • Constructor Details

    • 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 Details

    • getOp

      public Types.OpOpDnn getOp()
    • getOpString

      public String getOpString()
      Specified by:
      getOpString 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.
    • isMultiThreadedOpType

      public boolean isMultiThreadedOpType()
      Specified by:
      isMultiThreadedOpType in class MultiThreadedHop
    • constructLops

      public Lop constructLops()
      Specified by:
      constructLops in class Hop
    • setOp

      public void setOp(Types.OpOpDnn op)
    • isStride1Pad0

      public boolean isStride1Pad0()
    • constructDnnLops

      public Lop constructDnnLops(Types.ExecType et, List<Hop> inputs)
    • 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
    • clone

      public Object clone() throws CloneNotSupportedException
      Specified by:
      clone in class Hop
      Throws:
      CloneNotSupportedException
    • compare

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