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 |
numSArguments |
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) |
protected |
DynamicCustomOpsBuilder(String opName,
long hash,
int numInputs,
int numOutputs,
boolean inplaceAllowed,
int numTArguments,
int numIArguments,
int numSArguments) |
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.DynamicCustomOpsBuilder |
addStringArguments(List<String> sArgs)
This method takes arbitrary number of String arguments for op,
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
addStringArguments(String... sArgs)
This method takes arbitrary number of String arguments for op,
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
addStringArguments(String arg)
This method takes arbitrary number of String arguments for op,
Note that this ACCUMULATES arguments.
|
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 int numSArguments
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)
protected DynamicCustomOpsBuilder(String opName, long hash, int numInputs, int numOutputs, boolean inplaceAllowed, int numTArguments, int numIArguments, int numSArguments)
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 addStringArguments(List<String> sArgs)
sArgs
- public DynamicCustomOp.DynamicCustomOpsBuilder addStringArguments(String arg)
arg
- public DynamicCustomOp.DynamicCustomOpsBuilder addStringArguments(String... sArgs)
sArgs
- public DynamicCustomOp.DynamicCustomOpsBuilder addIntegerArguments(long arg)
arg
- public DynamicCustomOp.DynamicCustomOpsBuilder addIntegerArguments(int... iargs)
iargs
- public DynamicCustomOp.DynamicCustomOpsBuilder addBooleanArguments(boolean... bargs)
bargs
- public DynamicCustomOp.DynamicCustomOpsBuilder addFloatingPointArguments(Double... targs)
public DynamicCustomOp.DynamicCustomOpsBuilder addOutputShape(LongShapeDescriptor shape)
shape
- public DynamicCustomOp build()
public int getNumOutputs()
Copyright © 2022. All rights reserved.