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>gridprotected 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()protectedBaseGridOp(List<Op> ops)BaseGridOp(INDArray x, INDArray y)protectedBaseGridOp(GridPointers... pointers)protectedBaseGridOp(OpDescriptor... descriptors)protectedBaseGridOp(Op... ops)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description GridDescriptorgetGridDescriptor()StringonnxName()The opName of this function in onnxStringtensorflowName()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:
getGridDescriptorin interfaceGridOp- Returns:
-
onnxName
public String onnxName()
Description copied from class:DifferentialFunctionThe opName of this function in onnx
-
tensorflowName
public String tensorflowName()
Description copied from class:DifferentialFunctionThe opName of this function tensorflow- Overrides:
tensorflowNamein classBaseOp- Returns:
-
-