Package org.nd4j.imports.converters
Class DifferentialFunctionClassHolder
- java.lang.Object
-
- org.nd4j.imports.converters.DifferentialFunctionClassHolder
-
public class DifferentialFunctionClassHolder extends Object
-
-
Method Summary
-
-
-
Method Detail
-
getFieldsForFunction
public Map<String,Field> getFieldsForFunction(DifferentialFunction function)
Get the fields for a givenDifferentialFunction
- Parameters:
function
- the function to get the fields for- Returns:
- the fields for a given function
-
getOpDefByTensorflowName
public OpDef getOpDefByTensorflowName(String name)
Get the op definition of a given tensorflow op. Note that if the name does not exist, anND4JIllegalStateException
will be thrown- Parameters:
name
- the name of the op- Returns:
- the op definition for a given op
-
getOpDescriptorForOnnx
public OpDescriptor getOpDescriptorForOnnx(String name)
Get the op definition of a given onnx op Note that if the name does not exist, anND4JIllegalStateException
will be thrown.- Parameters:
name
- the name of the op- Returns:
- the op definition for a given op
-
getOpWithTensorflowName
public DifferentialFunction getOpWithTensorflowName(String tensorflowName)
Get the- Parameters:
tensorflowName
-- Returns:
-
getOpWithOnnxName
public DifferentialFunction getOpWithOnnxName(String onnxName)
-
missingTensorflowOps
public Set<String> missingTensorflowOps()
Returns the missing tensorflow ops- Returns:
-
hasName
public boolean hasName(String name)
- Parameters:
name
-- Returns:
-
getInstance
public DifferentialFunction getInstance(String name)
- Parameters:
name
-- Returns:
-
customOpClassForHashAndName
public Class<?> customOpClassForHashAndName(long customOpHash, String name)
-
getInstance
public static DifferentialFunctionClassHolder getInstance()
-
getTensorFlowNames
public Map<String,DifferentialFunction> getTensorFlowNames()
-
-