public class Cross extends DynamicCustomOp
DynamicCustomOp.DynamicCustomOpsBuilder, DynamicCustomOp.SameDiffBuilderiArguments, inplaceCall, inputArguments, outputArguments, outputVariables, tArgumentsdimensions, extraArgs, inPlace, sameDiff, scalarValue| Constructor and Description |
|---|
Cross() |
Cross(INDArray a,
INDArray b,
INDArray out) |
Cross(SameDiff sameDiff,
SDVariable[] args) |
| Modifier and Type | Method and Description |
|---|---|
java.util.List<SDVariable> |
doDiff(java.util.List<SDVariable> gradients)
The actual implementation for automatic differentiation.
|
java.lang.String |
opName()
This method returns op opName as string
|
java.lang.String |
tensorflowName()
The opName of this function tensorflow
|
addIArgument, addIArgument, addInputArgument, addOutputArgument, addTArgument, asProperties, assertValidForExecution, builder, calculateOutputShape, getDescriptor, getIArgument, getInputArgument, getOutputArgument, getTArgument, iArgs, initFromOnnx, initFromTensorFlow, inputArguments, numIArguments, numInputArguments, numOutputArguments, numTArguments, onnxName, opHash, opNum, opType, outputArguments, outputVariables, outputVariables, populateInputsAndOutputsFromSameDiff, removeIArgument, removeInputArgument, removeOutputArgument, removeTArgument, sameDiffBuilder, setInputArgument, setOutputArgument, tArgs, toStringarg, arg, argNames, args, attributeAdaptersForFunction, configFieldName, diff, dup, equals, f, getNumOutputs, getValue, hashCode, hasPlaceHolderInputs, isConfigProperties, larg, mappingsForFunction, onnxNames, outputVariable, outputVariablesNames, propertiesForFunction, rarg, resolvePropertiesFromSameDiffBeforeExecution, setInstanceId, setValueFor, tensorflowNamesclone, finalize, getClass, notify, notifyAll, wait, wait, waitisInplaceCallpublic Cross()
public Cross(SameDiff sameDiff, SDVariable[] args)
public java.lang.String opName()
DynamicCustomOpopName in interface CustomOpopName in class DynamicCustomOppublic java.lang.String tensorflowName()
DifferentialFunctiontensorflowName in class DynamicCustomOppublic java.util.List<SDVariable> doDiff(java.util.List<SDVariable> gradients)
DifferentialFunctiondoDiff in class DynamicCustomOp