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,
Types.ExecType et,
int k) |
DnnTransform(Lop input1,
Lop input2,
Types.OpOpDnn op,
Types.DataType dt,
Types.ValueType vt,
Types.ExecType et,
int k) |
DnnTransform(Lop input,
Types.OpOpDnn op,
Types.DataType dt,
Types.ValueType vt,
Types.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, setExecType, setFederatedOutput, setFilename, setPrivacyConstraint, setValueType, setVisitedpublic DnnTransform(Lop input, Types.OpOpDnn op, Types.DataType dt, Types.ValueType vt, Types.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, Types.ExecType et, int k)
public DnnTransform(Lop input1, Lop input2, Lop input3, Types.OpOpDnn op, Types.DataType dt, Types.ValueType vt, Types.ExecType et, int k)
public void updateLopProperties()
public String toString()
Loppublic Types.OpOpDnn getOp()
public String getInstructions(String input, String bias, String output)
LopgetInstructions in class Lopinput - input 1bias - input 2output - outputpublic String getInstructions(String input, String C, String HW, String output)
LopgetInstructions in class Lopinput - input 1C - input 2HW - input 3output - outputpublic String getInstructions(String input1, String input2, String input3, String input4, String output)
LopgetInstructions in class Lopinput1 - input 1input2 - input 2input3 - input 3input4 - input 4output - outputpublic String getInstructions(String[] inputs, String output)
getInstructions in class Loppublic String getInstructions(String input1, String input2, String input3, String input4, String input5, String input6, String output)
LopgetInstructions in class Lopinput1 - 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 © 2021 The Apache Software Foundation. All rights reserved.