org.nd4j.imports.converters

Class ImportClassMapping

java.lang.Object

org.nd4j.imports.converters.ImportClassMapping

public class ImportClassMapping
extends Object

A simple helper class for defining the following mappings: - Op name to op - Tensorflow op name to op (for import) - Onnx op name to op (for import)

Author:

Alex Black

Constructor Summary

Constructors

Constructor and Description
ImportClassMapping()

Method Summary

Modifier and Type	Method and Description
static Map<String,DifferentialFunction>	getOnnxOpMappingFunctions()
static List<Class<?>>	getOpClasses()
static Map<String,DifferentialFunction>	getOpNameMapping()
static Map<String,DifferentialFunction>	getTFOpMappingFunctions()

Methods inherited from class java.lang.Object

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

Constructor Detail

ImportClassMapping

public ImportClassMapping()

Method Detail

getOpClasses

public static List<Class<?>> getOpClasses()

getTFOpMappingFunctions

public static Map<String,DifferentialFunction> getTFOpMappingFunctions()

getOnnxOpMappingFunctions

public static Map<String,DifferentialFunction> getOnnxOpMappingFunctions()

getOpNameMapping

public static Map<String,DifferentialFunction> getOpNameMapping()