DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.DynamicCustomOpsBuilder.addBooleanArguments(boolean... bargs) |
This method takes arbitrary number of Integer arguments for op,
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.DynamicCustomOpsBuilder.addFloatingPointArguments(Double... targs) |
This method takes arbitrary number of Double arguments for op,
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.DynamicCustomOpsBuilder.addInputs(INDArray... inputs) |
This method
takes arbitrary number of input INDArrays in, as Op input
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.DynamicCustomOpsBuilder.addIntegerArguments(int... iargs) |
This method takes arbitrary number of Integer arguments for op,
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.DynamicCustomOpsBuilder.addIntegerArguments(long arg) |
This method takes arbitrary number of Integer arguments for op,
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.DynamicCustomOpsBuilder.addIntegerArguments(List<Integer> iargs) |
This method takes arbitrary number of Integer arguments for op,
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
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 |
DynamicCustomOp.DynamicCustomOpsBuilder.addOutputShape(LongShapeDescriptor shape) |
Adds an oup
|
DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.DynamicCustomOpsBuilder.addStringArguments(String arg) |
This method takes arbitrary number of String arguments for op,
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.DynamicCustomOpsBuilder.addStringArguments(String... sArgs) |
This method takes arbitrary number of String arguments for op,
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.DynamicCustomOpsBuilder.addStringArguments(List<String> sArgs) |
This method takes arbitrary number of String arguments for op,
Note that this ACCUMULATES arguments.
|
static DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.builder(String opName) |
This method takes custom opname, and return Op DynamicCustomOpsBuilder instance
|
DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.DynamicCustomOpsBuilder.callInplace(boolean reallyCall) |
Whether an op call is in place or not.
|