Class BaseGridOp
- java.lang.Object
-
- org.nd4j.autodiff.functions.DifferentialFunction
-
- org.nd4j.linalg.api.ops.BaseOp
-
- org.nd4j.linalg.api.ops.impl.grid.BaseGridOp
-
- Direct Known Subclasses:
BaseMetaOp
,FreeGridOp
public abstract class BaseGridOp extends BaseOp implements GridOp
-
-
Field Summary
Fields Modifier and Type Field Description protected List<GridPointers>
grid
protected List<OpDescriptor>
queuedOps
-
Fields inherited from class org.nd4j.linalg.api.ops.BaseOp
dimensionz, extraArgz, x, xVertexId, y, yVertexId, z, zVertexId
-
Fields inherited from class org.nd4j.autodiff.functions.DifferentialFunction
dimensions, extraArgs, inPlace, ownName, ownNameSetWithDefault, sameDiff, scalarValue
-
-
Constructor Summary
Constructors Modifier Constructor Description BaseGridOp()
protected
BaseGridOp(List<Op> ops)
BaseGridOp(INDArray x, INDArray y)
protected
BaseGridOp(GridPointers... pointers)
protected
BaseGridOp(OpDescriptor... descriptors)
protected
BaseGridOp(Op... ops)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description GridDescriptor
getGridDescriptor()
String
onnxName()
The opName of this function in onnxString
tensorflowName()
The opName of this function tensorflow-
Methods inherited from class org.nd4j.linalg.api.ops.BaseOp
clearArrays, computeVariables, defineDimensions, dimensions, equals, extraArgs, extraArgsBuff, extraArgsDataBuff, getFinalResult, getInputArgument, getNumOutputs, getOpType, hashCode, initFromOnnx, initFromTensorFlow, outputVariables, setX, setY, setZ, toCustomOp, toString, x, y, z
-
Methods inherited from class org.nd4j.autodiff.functions.DifferentialFunction
arg, arg, argNames, args, attributeAdaptersForFunction, calculateOutputDataTypes, calculateOutputShape, calculateOutputShape, configFieldName, configureWithSameDiff, diff, doDiff, dup, getBooleanFromProperty, getDoubleValueFromProperty, getIntValueFromProperty, getLongValueFromProperty, getStringFromProperty, getValue, isConfigProperties, larg, mappingsForFunction, onnxNames, opName, opNum, opType, outputs, outputVariable, outputVariables, outputVariablesNames, propertiesForFunction, rarg, replaceArg, setInstanceId, setPropertiesForFunction, setValueFor, tensorflowNames
-
Methods inherited from class java.lang.Object
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
-
Methods inherited from interface org.nd4j.linalg.api.ops.Op
clearArrays, extraArgs, extraArgsBuff, extraArgsDataBuff, opName, opNum, setExtraArgs, setX, setY, setZ, toCustomOp, x, y, z
-
-
-
-
Field Detail
-
queuedOps
protected List<OpDescriptor> queuedOps
-
grid
protected List<GridPointers> grid
-
-
Constructor Detail
-
BaseGridOp
public BaseGridOp()
-
BaseGridOp
protected BaseGridOp(Op... ops)
-
BaseGridOp
protected BaseGridOp(OpDescriptor... descriptors)
-
BaseGridOp
protected BaseGridOp(GridPointers... pointers)
-
-
Method Detail
-
getGridDescriptor
public GridDescriptor getGridDescriptor()
- Specified by:
getGridDescriptor
in interfaceGridOp
- Returns:
-
onnxName
public String onnxName()
Description copied from class:DifferentialFunction
The opName of this function in onnx
-
tensorflowName
public String tensorflowName()
Description copied from class:DifferentialFunction
The opName of this function tensorflow- Overrides:
tensorflowName
in classBaseOp
- Returns:
-
-