public interface CustomOp
Modifier and Type | Method and Description |
---|---|
void |
addIArgument(int... arg) |
void |
addInputArgument(INDArray... arg) |
void |
addOutputArgument(INDArray... arg) |
void |
addTArgument(double... arg) |
void |
assertValidForExecution()
Asserts a valid state for execution,
otherwise throws an
ND4JIllegalStateException |
List<int[]> |
calculateOutputShape()
Calculate the output shape for this op
|
CustomOpDescriptor |
getDescriptor()
Get the custom op descriptor if one is available.
|
Integer |
getIArgument(int index) |
INDArray |
getInputArgument(int index) |
INDArray |
getOutputArgument(int index) |
Double |
getTArgument(int index) |
int[] |
iArgs() |
INDArray[] |
inputArguments() |
boolean |
isInplaceCall()
This method returns true if op is supposed to be executed inplace
|
int |
numIArguments() |
int |
numInputArguments() |
int |
numOutputArguments() |
int |
numTArguments() |
long |
opHash()
This method returns LongHash of the opName()
|
String |
opName()
This method returns op opName as string
|
INDArray[] |
outputArguments() |
void |
populateInputsAndOutputsFromSameDiff()
Attempt to populate inputs and outputs from samediff.
|
void |
removeIArgument(Integer arg) |
void |
removeInputArgument(INDArray arg) |
void |
removeOutputArgument(INDArray arg) |
void |
removeTArgument(Double arg) |
double[] |
tArgs() |
String opName()
long opHash()
boolean isInplaceCall()
INDArray[] outputArguments()
INDArray[] inputArguments()
int[] iArgs()
double[] tArgs()
void addIArgument(int... arg)
void removeIArgument(Integer arg)
Integer getIArgument(int index)
int numIArguments()
void addTArgument(double... arg)
void removeTArgument(Double arg)
Double getTArgument(int index)
int numTArguments()
void addInputArgument(INDArray... arg)
void removeInputArgument(INDArray arg)
INDArray getInputArgument(int index)
int numInputArguments()
void addOutputArgument(INDArray... arg)
void removeOutputArgument(INDArray arg)
INDArray getOutputArgument(int index)
int numOutputArguments()
List<int[]> calculateOutputShape()
CustomOpDescriptor getDescriptor()
void assertValidForExecution()
ND4JIllegalStateException
void populateInputsAndOutputsFromSameDiff()
Copyright © 2018. All rights reserved.