public static class DynamicCustomOp.DynamicCustomOpsBuilder extends Object
Modifier and Type | Field and Description |
---|---|
protected boolean |
inplaceAllowed |
protected boolean |
inplaceCall |
protected int |
numBArguments |
protected int |
numIArguments |
protected int |
numInputs |
protected int |
numOutputs |
protected int |
numTArguments |
protected long |
opHash |
protected String |
opName |
protected List<LongShapeDescriptor> |
outputShapes |
Modifier | Constructor and Description |
---|---|
protected |
DynamicCustomOpsBuilder(String opName,
long hash,
int numInputs,
int numOutputs,
boolean inplaceAllowed,
int numTArguments,
int numIArguments) |
Modifier and Type | Method and Description |
---|---|
DynamicCustomOp.DynamicCustomOpsBuilder |
addBooleanArguments(boolean... bargs)
This method takes arbitrary number of Integer arguments for op,
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
addFloatingPointArguments(Double... targs)
This method takes arbitrary number of Double arguments for op,
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
addInputs(INDArray... inputs)
This method
takes arbitrary number of input INDArrays in, as Op input
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
addIntegerArguments(int... iargs)
This method takes arbitrary number of Integer arguments for op,
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
addIntegerArguments(List<Integer> iargs)
This method takes arbitrary number of Integer arguments for op,
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
addIntegerArguments(long arg)
This method takes arbitrary number of Integer arguments for op,
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
addOutputs(INDArray... outputs)
This method takes arbitrary number of
output INDArrays in, to store operation result
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
addOutputShape(LongShapeDescriptor shape)
Adds an oup
|
DynamicCustomOp |
build() |
DynamicCustomOp.DynamicCustomOpsBuilder |
callInplace(boolean reallyCall)
Whether an op call is in place or not.
|
int |
getNumOutputs() |
protected String opName
protected int numInputs
protected int numOutputs
protected int numTArguments
protected int numIArguments
protected int numBArguments
protected boolean inplaceCall
protected boolean inplaceAllowed
protected long opHash
protected List<LongShapeDescriptor> outputShapes
protected DynamicCustomOpsBuilder(String opName, long hash, int numInputs, int numOutputs, boolean inplaceAllowed, int numTArguments, int numIArguments)
public DynamicCustomOp.DynamicCustomOpsBuilder addInputs(INDArray... inputs)
inputs
- public DynamicCustomOp.DynamicCustomOpsBuilder addOutputs(INDArray... outputs)
outputs
- public DynamicCustomOp.DynamicCustomOpsBuilder callInplace(boolean reallyCall)
reallyCall
- public DynamicCustomOp.DynamicCustomOpsBuilder addIntegerArguments(List<Integer> iargs)
iargs
- public DynamicCustomOp.DynamicCustomOpsBuilder addIntegerArguments(long arg)
arg
- public DynamicCustomOp.DynamicCustomOpsBuilder addIntegerArguments(int... iargs)
iargs
- public DynamicCustomOp.DynamicCustomOpsBuilder addBooleanArguments(boolean... bargs)
iargs
- public DynamicCustomOp.DynamicCustomOpsBuilder addFloatingPointArguments(Double... targs)
public DynamicCustomOp.DynamicCustomOpsBuilder addOutputShape(LongShapeDescriptor shape)
shape
- public DynamicCustomOp build()
public int getNumOutputs()
Copyright © 2020. All rights reserved.