public abstract class BaseReduceOp extends BaseOp implements ReduceOp
Modifier and Type | Field and Description |
---|---|
protected boolean |
isComplex |
protected boolean |
isEmptyReduce |
protected boolean |
keepDims |
dimensionz, extraArgz, x, xVertexId, y, yVertexId, z, zVertexId
dimensions, extraArgs, inPlace, sameDiff, scalarValue
Constructor and Description |
---|
BaseReduceOp() |
BaseReduceOp(INDArray x,
boolean keepDims,
int... dimensions) |
BaseReduceOp(INDArray x,
INDArray y,
INDArray z,
boolean keepDims,
int[] dimensions) |
BaseReduceOp(INDArray x,
INDArray y,
INDArray z,
int... dimensions) |
BaseReduceOp(INDArray x,
INDArray y,
int... dimensions) |
BaseReduceOp(INDArray x,
int... dimensions) |
BaseReduceOp(SameDiff sameDiff) |
BaseReduceOp(SameDiff sameDiff,
SDVariable i_v) |
BaseReduceOp(SameDiff sameDiff,
SDVariable i_v,
int[] dimensions) |
BaseReduceOp(SameDiff sameDiff,
SDVariable i_v,
int[] dimensions,
boolean keepDims) |
BaseReduceOp(SameDiff sameDiff,
SDVariable i_v,
SDVariable i_v2,
int[] dimensions) |
BaseReduceOp(SameDiff sameDiff,
SDVariable i_v,
SDVariable i_v2,
int[] dimensions,
boolean keepDims) |
Modifier and Type | Method and Description |
---|---|
abstract List<LongShapeDescriptor> |
calculateOutputShape()
Calculate the output shape for this op
|
protected boolean |
hasReductionIndices(NodeDef nodeDef) |
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 |
boolean |
isComplexAccumulation() |
boolean |
isKeepDims()
This method returns TRUE if we're going to keep axis, FALSE otherwise
|
INDArray |
noOp()
Returns the no op version
of the input
Basically when a reduce can't happen (eg: sum(0) on a row vector)
you have a no op state for a given reduction.
|
void |
setDimensions(int... dimensions) |
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, configFieldName, diff, doDiff, dup, f, getValue, isConfigProperties, larg, mappingsForFunction, onnxNames, opName, opNum, opType, outputVariable, outputVariables, outputVariablesNames, propertiesForFunction, rarg, replaceArg, setInstanceId, setPropertiesForFunction, setValueFor, tensorflowNames
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
dimensions, getFinalResult, getOpType, resultType, validateDataTypes
clearArrays, extraArgs, extraArgsBuff, extraArgsDataBuff, opName, opNum, setExtraArgs, setX, setY, setZ, toCustomOp, x, y, z
protected boolean keepDims
protected boolean isComplex
protected boolean isEmptyReduce
public BaseReduceOp(SameDiff sameDiff, SDVariable i_v, int[] dimensions, boolean keepDims)
public BaseReduceOp(SameDiff sameDiff, SDVariable i_v, SDVariable i_v2, int[] dimensions, boolean keepDims)
public BaseReduceOp(SameDiff sameDiff, SDVariable i_v)
public BaseReduceOp(SameDiff sameDiff, SDVariable i_v, int[] dimensions)
public BaseReduceOp(SameDiff sameDiff, SDVariable i_v, SDVariable i_v2, int[] dimensions)
public BaseReduceOp()
public BaseReduceOp(INDArray x, INDArray y, INDArray z, boolean keepDims, int[] dimensions)
public BaseReduceOp(INDArray x, int... dimensions)
public BaseReduceOp(INDArray x, boolean keepDims, int... dimensions)
public BaseReduceOp(SameDiff sameDiff)
public INDArray noOp()
ReduceOp
public boolean isKeepDims()
ReduceOp
isKeepDims
in interface ReduceOp
public abstract List<LongShapeDescriptor> calculateOutputShape()
DifferentialFunction
calculateOutputShape
in class DifferentialFunction
public void initFromTensorFlow(NodeDef nodeDef, SameDiff initWith, Map<String,AttrValue> attributesForNode, GraphDef graph)
DifferentialFunction
NodeDef
initFromTensorFlow
in class BaseOp
protected boolean hasReductionIndices(NodeDef nodeDef)
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 isComplexAccumulation()
isComplexAccumulation
in interface ReduceOp
public void setDimensions(int... dimensions)
setDimensions
in interface ReduceOp
Copyright © 2019. All rights reserved.