public class DnnOp extends MultiThreadedHop
_beginColumn, _beginLine, _endColumn, _endLine, _filename, _text, CPThreshold
Constructor and Description |
---|
DnnOp(String l,
Types.DataType dt,
Types.ValueType vt,
Types.OpOpDnn o,
ArrayList<Hop> inp)
Create a hop from the builtin expression
|
Modifier and Type | Method and Description |
---|---|
boolean |
allowsAllExecTypes() |
void |
checkArity()
Check whether this Hop has a correct number of inputs.
|
Object |
clone() |
boolean |
compare(Hop that) |
Lop |
constructDnnLops(LopProperties.ExecType et,
ArrayList<Hop> inputs) |
Lop |
constructLops() |
Types.OpOpDnn |
getOp() |
String |
getOpString() |
boolean |
isGPUEnabled()
In memory-based optimizer mode (see OptimizerUtils.isMemoryBasedOptLevel()),
the exectype is determined by checking this method as well as memory budget of this Hop.
|
boolean |
isMultiThreadedOpType() |
boolean |
isStride1Pad0() |
void |
refreshSizeInformation()
Update the output size information for this hop.
|
void |
setOp(Types.OpOpDnn op) |
getMaxNumThreads, setMaxNumThreads
addAllInputs, addInput, checkAndSetForcedPlatform, checkAndSetInvalidCPDimsAndSize, clearMemEstimate, colsKnown, computeBoundsInformation, computeBoundsInformation, computeBoundsInformation, computeMemEstimate, computeSizeInformation, computeSizeInformation, computeSizeInformation, constructAndSetLopsDataFlowProperties, createOffsetLop, dimsKnown, dimsKnown, dimsKnownAny, getBeginColumn, getBeginLine, getBlocksize, getDataCharacteristics, getDataType, getDim, getDim1, getDim2, getEndColumn, getEndLine, getExecType, getFilename, getForcedExecType, getHopID, getInput, getInputMemEstimate, getInputOutputSize, getIntermediateMemEstimate, getLength, getLops, getMemEstimate, getName, getNnz, getOutputMemEstimate, getParent, getPrivacy, getSparsity, getSpBroadcastSize, getText, getUpdateType, getValueType, hasMatrixInputWithDifferentBlocksizes, hasValidCPDimsAndSize, isMatrix, isMemEstimated, isOutputEmptyBlocks, isScalar, isTransposeSafe, isVisited, printErrorLocation, refreshColsParameterInformation, refreshColsParameterInformation, refreshMemEstimates, refreshRowsParameterInformation, refreshRowsParameterInformation, requiresCheckpoint, requiresCompression, requiresLineageCaching, requiresReblock, requiresRecompile, resetExecType, resetRecompilationFlag, resetRecompilationFlag, resetVisitStatus, resetVisitStatus, resetVisitStatus, resetVisitStatusForced, rowsKnown, setBeginColumn, setBeginLine, setBlocksize, setDataType, setDim, setDim1, setDim2, setEndColumn, setEndLine, setFilename, setForcedExecType, setLops, setMemEstimate, setName, setNnz, setOutputEmptyBlocks, setParseInfo, setPrivacy, setRequiresCheckpoint, setRequiresCompression, setRequiresLineageCaching, setRequiresReblock, setRequiresRecompile, setText, setUpdateType, setValueType, setVisited, setVisited
equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
ctxAndFilenameToParseInfo
public DnnOp(String l, Types.DataType dt, Types.ValueType vt, Types.OpOpDnn o, ArrayList<Hop> inp)
l
- name of the hopdt
- datatype (only supports matrix datatype)vt
- valuetype (only supports matrix valuetype)o
- type of this hopinp
- input hopspublic void checkArity()
Hop
checkArity
in class Hop
public Types.OpOpDnn getOp()
public String getOpString()
getOpString
in class Hop
public boolean isGPUEnabled()
Hop
isGPUEnabled
in class Hop
public boolean isMultiThreadedOpType()
isMultiThreadedOpType
in class MultiThreadedHop
public Lop constructLops()
constructLops
in class Hop
public void setOp(Types.OpOpDnn op)
public boolean isStride1Pad0()
public Lop constructDnnLops(LopProperties.ExecType et, ArrayList<Hop> inputs)
public boolean allowsAllExecTypes()
allowsAllExecTypes
in class Hop
public void refreshSizeInformation()
Hop
refreshSizeInformation
in class Hop
public Object clone() throws CloneNotSupportedException
clone
in class Hop
CloneNotSupportedException
Copyright © 2020 The Apache Software Foundation. All rights reserved.