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