org.nd4j.imports.converters

Class DifferentialFunctionClassHolder

java.lang.Object

org.nd4j.imports.converters.DifferentialFunctionClassHolder

public class DifferentialFunctionClassHolder
extends Object

Method Summary

Modifier and Type	Method and Description
Class<?>	customOpClassForHashAndName(long customOpHash, String name)
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
Map<String,DifferentialFunction>	getTensorFlowNames()
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()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

getFieldsForFunction

public Map<String,Field> getFieldsForFunction(DifferentialFunction function)

Get the fields for a given DifferentialFunction

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, an ND4JIllegalStateException 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, an ND4JIllegalStateException 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)

missingOnnxOps

public Set<String> missingOnnxOps()

Returns the missing onnx ops

Returns:

missingTensorflowOps

public Set<String> missingTensorflowOps()

Returns the missing tensorflow ops

Returns:

missingOps

public List<String> missingOps()

Returns the missing ops for c++ vs java.

Returns:

hasName

public boolean hasName(String name)

Parameters:

name -

Returns:

opNames

public Set<String> opNames()

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()