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(Types.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, setMaxNumThreadsaddAllInputs, 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, getInput, getInputMemEstimate, getInputOutputSize, getIntermediateMemEstimate, getLength, getLops, getMemEstimate, getName, getNnz, getOutputMemEstimate, getParent, getPrivacy, getSparsity, getSpBroadcastSize, getText, getUpdateType, getValueType, hasFederatedOutput, hasMatrixInputWithDifferentBlocksizes, hasValidCPDimsAndSize, isFederated, isFederatedOutput, 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, setRequiresDeCompression, setRequiresLineageCaching, setRequiresReblock, setRequiresRecompile, setText, setUpdateType, setValueType, setVisited, setVisited, toStringequals, getClass, hashCode, notify, notifyAll, wait, wait, waitctxAndFilenameToParseInfopublic 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()
HopcheckArity in class Hoppublic Types.OpOpDnn getOp()
public String getOpString()
getOpString in class Hoppublic boolean isGPUEnabled()
HopisGPUEnabled in class Hoppublic boolean isMultiThreadedOpType()
isMultiThreadedOpType in class MultiThreadedHoppublic Lop constructLops()
constructLops in class Hoppublic void setOp(Types.OpOpDnn op)
public boolean isStride1Pad0()
public Lop constructDnnLops(Types.ExecType et, ArrayList<Hop> inputs)
public boolean allowsAllExecTypes()
allowsAllExecTypes in class Hoppublic void refreshSizeInformation()
HoprefreshSizeInformation in class Hoppublic Object clone() throws CloneNotSupportedException
clone in class HopCloneNotSupportedExceptionCopyright © 2021 The Apache Software Foundation. All rights reserved.