public class ExternalErrorsFunction extends DynamicCustomOp
DynamicCustomOp.DynamicCustomOpsBuilder| Modifier and Type | Field and Description |
|---|---|
static String |
OP_NAME |
axis, bArguments, dArguments, iArguments, inplaceCall, inputArguments, outputArguments, outputVariables, tArgumentsdimensions, extraArgs, inPlace, ownName, ownNameSetWithDefault, sameDiff, scalarValue| Constructor and Description |
|---|
ExternalErrorsFunction() |
ExternalErrorsFunction(SameDiff sd,
List<SDVariable> inputs,
Map<String,INDArray> gradients) |
| Modifier and Type | Method and Description |
|---|---|
List<LongShapeDescriptor> |
calculateOutputShape()
Calculate the output shape for this op
|
List<LongShapeDescriptor> |
calculateOutputShape(OpContext oc)
Calculate the output shape for this op
|
List<SDVariable> |
doDiff(List<SDVariable> f1)
The actual implementation for automatic differentiation.
|
String |
getGradPlaceholderName() |
void |
initFromOnnx(Onnx.NodeProto node,
SameDiff initWith,
Map<String,Onnx.AttributeProto> attributesForNode,
Onnx.GraphProto graph)
Iniitialize the function from the given
Onnx.NodeProto |
void |
initFromTensorFlow(NodeDef nodeDef,
SameDiff initWith,
Map<String,AttrValue> attributesForNode,
GraphDef graph)
Initialize the function from the given
NodeDef |
String |
onnxName()
The opName of this function in onnx
|
String |
opName()
This method returns op opName as string
|
Op.Type |
opType()
The type of the op
|
SDVariable[] |
outputVariables(String baseName)
Return the output functions for this differential function.
|
String |
tensorflowName()
The opName of this function tensorflow
|
String |
toString() |
void |
updateBeforeExecution() |
addBArgument, addDArgument, addIArgument, addIArgument, addInputArgument, addOutputArgument, addTArgument, assertValidForExecution, bArgs, builder, clearArrays, dArgs, getBArgument, getDescriptor, getIArgument, getInputArgument, getOutputArgument, getTArgument, iArgs, inputArguments, numBArguments, numDArguments, numIArguments, numInputArguments, numOutputArguments, numTArguments, opHash, opNum, outputArguments, outputVariables, removeIArgument, removeInputArgument, removeOutputArgument, removeTArgument, setInputArgument, setInputArguments, setOutputArgument, tArgs, wrapFilterNull, wrapOrNull, wrapOrNullarg, arg, argNames, args, attributeAdaptersForFunction, calculateOutputDataTypes, configFieldName, diff, dup, equals, getNumOutputs, getValue, hashCode, isConfigProperties, larg, mappingsForFunction, onnxNames, outputs, outputVariable, outputVariablesNames, propertiesForFunction, rarg, replaceArg, setInstanceId, setPropertiesForFunction, setValueFor, tensorflowNamesclone, finalize, getClass, notify, notifyAll, wait, wait, waitisInplaceCallpublic static final String OP_NAME
public ExternalErrorsFunction(SameDiff sd, List<SDVariable> inputs, Map<String,INDArray> gradients)
public ExternalErrorsFunction()
public String getGradPlaceholderName()
public SDVariable[] outputVariables(String baseName)
DifferentialFunctionoutputVariables in class DynamicCustomOppublic List<SDVariable> doDiff(List<SDVariable> f1)
DifferentialFunctiondoDiff in class DynamicCustomOppublic void updateBeforeExecution()
public void initFromTensorFlow(NodeDef nodeDef, SameDiff initWith, Map<String,AttrValue> attributesForNode, GraphDef graph)
DifferentialFunctionNodeDefinitFromTensorFlow in class DynamicCustomOppublic void initFromOnnx(Onnx.NodeProto node, SameDiff initWith, Map<String,Onnx.AttributeProto> attributesForNode, Onnx.GraphProto graph)
DifferentialFunctionOnnx.NodeProtoinitFromOnnx in class DynamicCustomOppublic String onnxName()
DifferentialFunctiononnxName in class DynamicCustomOppublic String tensorflowName()
DifferentialFunctiontensorflowName in class DynamicCustomOppublic String opName()
DynamicCustomOpopName in interface CustomOpopName in class DynamicCustomOppublic String toString()
toString in class DynamicCustomOppublic List<LongShapeDescriptor> calculateOutputShape()
DifferentialFunctioncalculateOutputShape in interface CustomOpcalculateOutputShape in class DynamicCustomOppublic List<LongShapeDescriptor> calculateOutputShape(OpContext oc)
CustomOpcalculateOutputShape in interface CustomOpcalculateOutputShape in class DynamicCustomOppublic Op.Type opType()
DifferentialFunctionopType in class DynamicCustomOpCopyright © 2021. All rights reserved.