org.deeplearning4j.nn.layers

Class HelperUtils

java.lang.Object

org.deeplearning4j.nn.layers.HelperUtils

public class HelperUtils
extends Object

Simple meta helper util class for instantiating platform specific layer helpers that handle interaction with lower level libraries like cudnn and onednn.

Author:

Adam Gibson

Constructor Summary

Constructors

Constructor and Description
HelperUtils()

Method Summary

Modifier and Type	Method and Description
static <T extends LayerHelper> T	createHelper(String cudnnHelperClassName, String oneDnnClassName, Class<? extends LayerHelper> layerHelperSuperClass, String layerName, Object... arguments) Creates a LayerHelper for use with platform specific code.

Methods inherited from class java.lang.Object

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

Constructor Detail

HelperUtils

public HelperUtils()

Method Detail

createHelper

public static <T extends LayerHelper> T createHelper(String cudnnHelperClassName,
                                                     String oneDnnClassName,
                                                     Class<? extends LayerHelper> layerHelperSuperClass,
                                                     String layerName,
                                                     Object... arguments)

Creates a LayerHelper for use with platform specific code.

Type Parameters:

T - the actual class type to be returned

Parameters:

cudnnHelperClassName - the cudnn class name

oneDnnClassName - the one dnn class name

layerHelperSuperClass - the layer helper super class

layerName - the name of the layer to be created

arguments - the arguments to be used in creation of the layer

Returns: