Package | Description |
---|---|
org.nd4j.linalg.api.ops |
Modifier and Type | Method and Description |
---|---|
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(List<Integer> 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.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(List<String> sArgs)
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(String arg)
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.
|
Copyright © 2022. All rights reserved.