io.mxnet.caffetranslator

Class GenerationHelper

java.lang.Object

io.mxnet.caffetranslator.GenerationHelper

public class GenerationHelper
extends java.lang.Object

Field Summary

Fields

Modifier and Type	Field and Description
protected org.stringtemplate.v4.STGroup	stGroupDir
protected org.stringtemplate.v4.STGroup	stGroupFile

Constructor Summary

Constructors

Constructor and Description
GenerationHelper()

Method Summary

Modifier and Type	Method and Description
void	fillNameDataAndVar(org.stringtemplate.v4.ST st, Layer layer)
java.lang.String	generateVar(java.lang.String varName, java.lang.String symName, java.lang.String lr_mult, java.lang.String wd_mult, java.lang.String init, java.util.List<java.lang.Integer> shape)
java.lang.String	getInit(java.lang.String fillerType, java.lang.String fillerValue)
org.stringtemplate.v4.ST	getTemplate(java.lang.String name)
java.lang.String	getVarname(java.lang.String name)
java.util.List<java.lang.String>	getVarNames(java.util.List<java.lang.String> names)
java.lang.String	initializeParam(java.lang.String varname, int childIndex, java.lang.String initializer)
GeneratorOutput	makeGeneratorOutput(java.lang.String code, int numLayersTranslated)
void	simpleFillTemplate(org.stringtemplate.v4.ST st, java.lang.String name, Layer layer, java.lang.String key, java.lang.String defaultValue, java.lang.String... altKeys)

Methods inherited from class java.lang.Object

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

Field Detail

stGroupDir

protected final org.stringtemplate.v4.STGroup stGroupDir

stGroupFile

protected final org.stringtemplate.v4.STGroup stGroupFile

Constructor Detail

GenerationHelper

public GenerationHelper()

Method Detail

getTemplate

public org.stringtemplate.v4.ST getTemplate(java.lang.String name)

generateVar

public java.lang.String generateVar(java.lang.String varName,
                                    java.lang.String symName,
                                    java.lang.String lr_mult,
                                    java.lang.String wd_mult,
                                    java.lang.String init,
                                    java.util.List<java.lang.Integer> shape)

getInit

public java.lang.String getInit(java.lang.String fillerType,
                                java.lang.String fillerValue)

getVarname

public java.lang.String getVarname(java.lang.String name)

getVarNames

public java.util.List<java.lang.String> getVarNames(java.util.List<java.lang.String> names)

fillNameDataAndVar

public void fillNameDataAndVar(org.stringtemplate.v4.ST st,
                               Layer layer)

simpleFillTemplate

public void simpleFillTemplate(org.stringtemplate.v4.ST st,
                               java.lang.String name,
                               Layer layer,
                               java.lang.String key,
                               java.lang.String defaultValue,
                               java.lang.String... altKeys)

makeGeneratorOutput

public GeneratorOutput makeGeneratorOutput(java.lang.String code,
                                           int numLayersTranslated)

initializeParam

public java.lang.String initializeParam(java.lang.String varname,
                                        int childIndex,
                                        java.lang.String initializer)