Package 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 Description HelperUtils()
-
Method Summary
All Methods Static Methods Concrete Methods Modifier and Type Method Description static <T extends LayerHelper>
TcreateHelper(String cudnnHelperClassName, String oneDnnClassName, Class<? extends LayerHelper> layerHelperSuperClass, String layerName, Object... arguments)
Creates aLayerHelper
for use with platform specific code.
-
-
-
Method Detail
-
createHelper
public static <T extends LayerHelper> T createHelper(String cudnnHelperClassName, String oneDnnClassName, Class<? extends LayerHelper> layerHelperSuperClass, String layerName, Object... arguments)
Creates aLayerHelper
for use with platform specific code.- Type Parameters:
T
- the actual class type to be returned- Parameters:
cudnnHelperClassName
- the cudnn class nameoneDnnClassName
- the one dnn class namelayerHelperSuperClass
- the layer helper super classlayerName
- the name of the layer to be createdarguments
- the arguments to be used in creation of the layer- Returns:
-
-