public class DifferentialFunctionClassHolder extends Object
Modifier and Type | Method and Description |
---|---|
Map<String,Field> |
getFieldsForFunction(DifferentialFunction function)
Get the fields for a given
DifferentialFunction |
static DifferentialFunctionClassHolder |
getInstance() |
DifferentialFunction |
getInstance(String name) |
OpDef |
getOpDefByTensorflowName(String name)
Get the op definition of a given
tensorflow op.
|
OpDescriptor |
getOpDescriptorForOnnx(String name)
Get the op definition of a given
onnx op
Note that if the name does not exist,
an
ND4JIllegalStateException
will be thrown. |
DifferentialFunction |
getOpWithOnnxName(String onnxName) |
DifferentialFunction |
getOpWithTensorflowName(String tensorflowName)
Get the
|
boolean |
hasName(String name) |
Set<String> |
missingOnnxOps()
Returns the missing onnx ops
|
List<String> |
missingOps()
Returns the missing ops
for c++ vs java.
|
Set<String> |
missingTensorflowOps()
Returns the missing tensorflow ops
|
Set<String> |
opNames() |
public Map<String,Field> getFieldsForFunction(DifferentialFunction function)
DifferentialFunction
function
- the function to get the fields forpublic OpDef getOpDefByTensorflowName(String name)
ND4JIllegalStateException
will be thrownname
- the name of the oppublic OpDescriptor getOpDescriptorForOnnx(String name)
ND4JIllegalStateException
will be thrown.name
- the name of the oppublic DifferentialFunction getOpWithTensorflowName(String tensorflowName)
tensorflowName
- public DifferentialFunction getOpWithOnnxName(String onnxName)
public Set<String> missingTensorflowOps()
public boolean hasName(String name)
name
- public DifferentialFunction getInstance(String name)
name
- public static DifferentialFunctionClassHolder getInstance()
Copyright © 2018. All rights reserved.