DynamicCustomOp.DynamicCustomOpsBuilder, DynamicCustomOp.SameDiffBuilder
permuteDims
inplaceCall, outputVariables
dimensions, extraArgs, inPlace, sameDiff, scalarValue
Constructor and Description |
---|
Permute() |
Permute(SameDiff sameDiff,
SDVariable i_v,
int[] permuteDims) |
Modifier and Type | Method and Description |
---|---|
List<SDVariable> |
doDiff(List<SDVariable> i_v)
The actual implementation for automatic differentiation.
|
String |
opName()
This method returns op opName as string
|
calculateOutputShape, initFromOnnx, initFromTensorFlow, mappingsForFunction, onnxName, propertiesForFunction, resolvePropertiesFromSameDiffBeforeExecution, tensorflowName
addIArgument, addInputArgument, addOutputArgument, addTArgument, asProperties, assertValidForExecution, builder, getDescriptor, getIArgument, getInputArgument, getOutputArgument, getTArgument, iArgs, inputArguments, numIArguments, numInputArguments, numOutputArguments, numTArguments, opHash, opNum, opType, outputArguments, outputVariables, outputVariables, populateInputsAndOutputsFromSameDiff, removeIArgument, removeInputArgument, removeOutputArgument, removeTArgument, sameDiffBuilder, setInputArgument, setOutputArgument, tArgs, toString, updateInputsFromSameDiff
arg, args, attributeAdaptersForFunction, configFieldName, diff, dup, equals, f, getValue, hashCode, hasPlaceHolderInputs, isConfigProperties, larg, onnxNames, rarg, setInstanceId, setValueFor, tensorflowNames
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
isInplaceCall
public Permute(SameDiff sameDiff, SDVariable i_v, int[] permuteDims)
public Permute()
public String opName()
DynamicCustomOp
public List<SDVariable> doDiff(List<SDVariable> i_v)
DifferentialFunction
Copyright © 2018. All rights reserved.