public static class DynamicCustomOp.SameDiffBuilder extends DynamicCustomOp.DynamicCustomOpsBuilder
inplaceAllowed, inplaceCall, numBArguments, numIArguments, numInputs, numOutputs, numTArguments, opHash, opName, outputShapes
Modifier | Constructor and Description |
---|---|
protected |
SameDiffBuilder(String opName,
SameDiff sameDiff,
long hash,
int numInputs,
int numOutputs,
boolean inplaceAllowed,
int numTArguments,
int numIArguments) |
Modifier and Type | Method and Description |
---|---|
DynamicCustomOp.DynamicCustomOpsBuilder |
addInputs(DifferentialFunction... inputs) |
DynamicCustomOp.DynamicCustomOpsBuilder |
addInputs(INDArray... inputs)
This method
takes arbitrary number of input INDArrays in, as Op input
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
addOutputs(DifferentialFunction... outputs) |
DynamicCustomOp.DynamicCustomOpsBuilder |
addOutputs(INDArray... outputs)
This method takes arbitrary number of
output INDArrays in, to store operation result
Note that this ACCUMULATES arguments.
|
DynamicCustomOp |
build() |
DynamicCustomOp.SameDiffBuilder |
sameDiff(SameDiff sameDiff) |
addBooleanArguments, addFloatingPointArguments, addIntegerArguments, addIntegerArguments, addIntegerArguments, addOutputShape, callInplace, getNumOutputs
public DynamicCustomOp.SameDiffBuilder sameDiff(SameDiff sameDiff)
public DynamicCustomOp.DynamicCustomOpsBuilder addInputs(INDArray... inputs)
DynamicCustomOp.DynamicCustomOpsBuilder
addInputs
in class DynamicCustomOp.DynamicCustomOpsBuilder
public DynamicCustomOp.DynamicCustomOpsBuilder addOutputs(INDArray... outputs)
DynamicCustomOp.DynamicCustomOpsBuilder
addOutputs
in class DynamicCustomOp.DynamicCustomOpsBuilder
public DynamicCustomOp.DynamicCustomOpsBuilder addInputs(DifferentialFunction... inputs)
public DynamicCustomOp.DynamicCustomOpsBuilder addOutputs(DifferentialFunction... outputs)
public DynamicCustomOp build()
build
in class DynamicCustomOp.DynamicCustomOpsBuilder
Copyright © 2019. All rights reserved.