public abstract class BaseBroadcastBoolOp extends BaseOp implements BroadcastOp
Modifier and Type | Field and Description |
---|---|
protected int[] |
dimension |
dimensionz, extraArgz, x, xVertexId, y, yVertexId, z, zVertexId
dimensions, extraArgs, inPlace, ownName, ownNameSetWithDefault, sameDiff, scalarValue
Constructor and Description |
---|
BaseBroadcastBoolOp(INDArray x,
INDArray y,
INDArray z,
int... dimension) |
BaseBroadcastBoolOp(SameDiff sameDiff) |
BaseBroadcastBoolOp(SameDiff sameDiff,
SDVariable i_v,
int[] dimension,
boolean inPlace) |
BaseBroadcastBoolOp(SameDiff sameDiff,
SDVariable i_v,
int[] dimension,
Object[] extraArgs) |
BaseBroadcastBoolOp(SameDiff sameDiff,
SDVariable i_v,
long[] shape,
boolean inPlace,
int[] dimension,
Object[] extraArgs) |
BaseBroadcastBoolOp(SameDiff sameDiff,
SDVariable i_v1,
SDVariable i_v2,
boolean inPlace,
int[] dimension) |
BaseBroadcastBoolOp(SameDiff sameDiff,
SDVariable i_v1,
SDVariable i_v2,
int[] dimension) |
BaseBroadcastBoolOp(SameDiff sameDiff,
SDVariable i_v1,
SDVariable i_v2,
int[] dimension,
Object[] extraArgs) |
Modifier and Type | Method and Description |
---|---|
List<LongShapeDescriptor> |
calculateOutputShape()
Calculate the output shape for this op
|
int[] |
getDimension()
Dimension to do the vector op along.
|
Op.Type |
getOpType() |
void |
initFromOnnx(Onnx.NodeProto node,
SameDiff initWith,
Map<String,Onnx.AttributeProto> attributesForNode,
Onnx.GraphProto graph)
Iniitialize the function from the given
Onnx.NodeProto |
void |
initFromTensorFlow(NodeDef nodeDef,
SameDiff initWith,
Map<String,AttrValue> attributesForNode,
GraphDef graph)
Initialize the function from the given
NodeDef |
Op.Type |
opType()
The type of the op
|
void |
setDimension(int... dimension)
Set the dimension for the vector op.
|
boolean |
validateDataTypes(boolean experimentalMode) |
clearArrays, defineDimensions, dimensions, equals, extraArgs, extraArgsBuff, extraArgsDataBuff, getFinalResult, getInputArgument, getNumOutputs, getOpType, hashCode, onnxName, outputVariables, setX, setY, setZ, tensorflowName, toCustomOp, toString, x, y, z
arg, arg, argNames, args, attributeAdaptersForFunction, calculateOutputDataTypes, calculateOutputShape, configFieldName, diff, doDiff, dup, getValue, isConfigProperties, larg, mappingsForFunction, onnxNames, opName, opNum, outputs, outputVariable, outputVariables, outputVariablesNames, propertiesForFunction, rarg, replaceArg, setInstanceId, setPropertiesForFunction, setValueFor, tensorflowNames
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
dimensions
clearArrays, extraArgs, extraArgsBuff, extraArgsDataBuff, opName, opNum, setExtraArgs, setX, setY, setZ, toCustomOp, x, y, z
public BaseBroadcastBoolOp(SameDiff sameDiff, SDVariable i_v1, SDVariable i_v2, int[] dimension)
public BaseBroadcastBoolOp(SameDiff sameDiff, SDVariable i_v1, SDVariable i_v2, boolean inPlace, int[] dimension)
public BaseBroadcastBoolOp(SameDiff sameDiff)
public BaseBroadcastBoolOp(SameDiff sameDiff, SDVariable i_v1, SDVariable i_v2, int[] dimension, Object[] extraArgs)
public BaseBroadcastBoolOp(SameDiff sameDiff, SDVariable i_v, int[] dimension, boolean inPlace)
public BaseBroadcastBoolOp(SameDiff sameDiff, SDVariable i_v, long[] shape, boolean inPlace, int[] dimension, Object[] extraArgs)
public BaseBroadcastBoolOp(SameDiff sameDiff, SDVariable i_v, int[] dimension, Object[] extraArgs)
public Op.Type opType()
DifferentialFunction
opType
in class DifferentialFunction
public List<LongShapeDescriptor> calculateOutputShape()
calculateOutputShape
in class DifferentialFunction
public int[] getDimension()
BroadcastOp
getDimension
in interface BroadcastOp
public void setDimension(int... dimension)
BroadcastOp
setDimension
in interface BroadcastOp
public void initFromTensorFlow(NodeDef nodeDef, SameDiff initWith, Map<String,AttrValue> attributesForNode, GraphDef graph)
DifferentialFunction
NodeDef
initFromTensorFlow
in class BaseOp
public void initFromOnnx(Onnx.NodeProto node, SameDiff initWith, Map<String,Onnx.AttributeProto> attributesForNode, Onnx.GraphProto graph)
DifferentialFunction
Onnx.NodeProto
initFromOnnx
in class BaseOp
public boolean validateDataTypes(boolean experimentalMode)
validateDataTypes
in interface BroadcastOp
public Op.Type getOpType()
getOpType
in interface BroadcastOp
Copyright © 2021. All rights reserved.