public class DnnTransform extends Lop
Lop.SimpleInstType, Lop.Type, Lop.VisitStatus
_beginColumn, _beginLine, _endColumn, _endLine, _filename, CP_CHILD_THREAD, CP_ROOT_THREAD_ID, DATATYPE_PREFIX, FILE_SEPARATOR, FRAME_VAR_NAME_PREFIX, INSTRUCTION_DELIMITOR, LITERAL_PREFIX, MATRIX_VAR_NAME_PREFIX, NAME_VALUE_SEPARATOR, OPERAND_DELIMITOR, PROCESS_PREFIX, SAMPLE_FRACTION, SCALAR_VAR_NAME_PREFIX, UPDATE_INPLACE_PREFIX, VALUETYPE_PREFIX, VARIABLE_NAME_PLACEHOLDER
Constructor and Description |
---|
DnnTransform(Lop input1,
Lop input2,
Lop input3,
Types.OpOpDnn op,
Types.DataType dt,
Types.ValueType vt,
LopProperties.ExecType et,
int k) |
DnnTransform(Lop input1,
Lop input2,
Types.OpOpDnn op,
Types.DataType dt,
Types.ValueType vt,
LopProperties.ExecType et,
int k) |
DnnTransform(Lop input,
Types.OpOpDnn op,
Types.DataType dt,
Types.ValueType vt,
LopProperties.ExecType et,
int k,
double intermediateMemBudget)
Constructor when we have one input.
|
Modifier and Type | Method and Description |
---|---|
void |
appendOpcode(StringBuilder sb) |
void |
appendOperands(int startInputIndex,
int endInputIndex,
String output,
StringBuilder sb) |
String |
getInstructions(String[] inputs,
String output) |
String |
getInstructions(String input,
String bias,
String output)
Method should be overridden if needed
|
String |
getInstructions(String input,
String C,
String HW,
String output)
Method should be overridden if needed
|
String |
getInstructions(String input1,
String input2,
String input3,
String input4,
String output)
Method should be overridden if needed
|
String |
getInstructions(String input1,
String input2,
String input3,
String input4,
String input5,
String input6,
String output)
Method should be overridden if needed
|
Types.OpOpDnn |
getOp()
method to get operation type
|
String |
toString()
Method to have Lops print their state.
|
void |
updateLopProperties() |
addInput, addOutput, addToDag, createReachable, getBeginColumn, getBeginLine, getDataType, getEndColumn, getEndLine, getExecType, getFilename, getID, getInputs, getInstructions, getInstructions, getInstructions, getInstructions, getInstructions, getInstructions, getInstructions, getLevel, getOutputParameters, getOutputs, getPrivacyConstraint, getProducesIntermediateOutput, getReachable, getSimpleInstructionType, getType, getValueType, getVisited, isDataExecLocation, isVariable, prepInputOperand, prepInputOperand, prepOperand, prepOperand, prepOutputOperand, prepOutputOperand, prepOutputOperand, prepScalarInputOperand, prepScalarInputOperand, prepScalarLabel, prepScalarOperand, printErrorLocation, removeConsumer, resetVisitStatus, setAllPositions, setBeginColumn, setBeginLine, setConsumerCount, setDataType, setEndColumn, setEndLine, setFilename, setPrivacyConstraint, setValueType, setVisited
public DnnTransform(Lop input, Types.OpOpDnn op, Types.DataType dt, Types.ValueType vt, LopProperties.ExecType et, int k, double intermediateMemBudget)
input
- low-level operatorop
- convolution transform operation typedt
- data typevt
- value typeet
- execution typek
- number of threadsintermediateMemBudget
- intermediate memory budgetpublic DnnTransform(Lop input1, Lop input2, Types.OpOpDnn op, Types.DataType dt, Types.ValueType vt, LopProperties.ExecType et, int k)
public DnnTransform(Lop input1, Lop input2, Lop input3, Types.OpOpDnn op, Types.DataType dt, Types.ValueType vt, LopProperties.ExecType et, int k)
public void updateLopProperties()
public String toString()
Lop
public Types.OpOpDnn getOp()
public String getInstructions(String input, String bias, String output)
Lop
getInstructions
in class Lop
input
- input 1bias
- input 2output
- outputpublic String getInstructions(String input, String C, String HW, String output)
Lop
getInstructions
in class Lop
input
- input 1C
- input 2HW
- input 3output
- outputpublic String getInstructions(String input1, String input2, String input3, String input4, String output)
Lop
getInstructions
in class Lop
input1
- input 1input2
- input 2input3
- input 3input4
- input 4output
- outputpublic String getInstructions(String[] inputs, String output)
getInstructions
in class Lop
public String getInstructions(String input1, String input2, String input3, String input4, String input5, String input6, String output)
Lop
getInstructions
in class Lop
input1
- input 1input2
- input 2input3
- input 3input4
- input 4input5
- input 5input6
- input 6output
- outputpublic void appendOpcode(StringBuilder sb)
public void appendOperands(int startInputIndex, int endInputIndex, String output, StringBuilder sb)
Copyright © 2020 The Apache Software Foundation. All rights reserved.