Interface GraphBuilderModule
-
public interface GraphBuilderModule
-
-
Method Summary
All Methods Instance Methods Abstract Methods Modifier and Type Method Description String
getModuleName()
A module should return its name.ComputationGraphConfiguration.GraphBuilder
updateBuilder(ComputationGraphConfiguration.GraphBuilder graph, String layerName, int inputSize, int[][] config, String inputLayer)
Add a layer to the collection of layers being generated by this module.
-
-
-
Method Detail
-
getModuleName
String getModuleName()
A module should return its name. Useful for automatically generating layer names.- Returns:
- The module name as a string.
-
updateBuilder
ComputationGraphConfiguration.GraphBuilder updateBuilder(ComputationGraphConfiguration.GraphBuilder graph, String layerName, int inputSize, int[][] config, String inputLayer)
Add a layer to the collection of layers being generated by this module.- Parameters:
graph
-layerName
-inputSize
-config
-inputLayer
-- Returns:
- An instance of GraphBuilder with appended layers.
-
-