io.mxnet.caffetranslator.generators

Class BaseGenerator

java.lang.Object

io.mxnet.caffetranslator.generators.BaseGenerator

All Implemented Interfaces:

SymbolGenerator

Direct Known Subclasses:

BatchNormGenerator, ConcatGenerator, ConvolutionGenerator, DeconvolutionGenerator, DropoutGenerator, EltwiseGenerator, FCGenerator, FlattenGenerator, PermuteGenerator, PluginIntLayerGenerator, PluginLossGenerator, PoolingGenerator, PowerGenerator, ReluGenerator, ScaleGenerator, SoftmaxOutputGenerator

public abstract class BaseGenerator
extends java.lang.Object
implements SymbolGenerator

Field Summary

Fields

Modifier and Type	Field and Description
protected GenerationHelper	gh

Constructor Summary

Constructors

Constructor and Description
BaseGenerator()

Method Summary

Modifier and Type	Method and Description
protected 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)
protected org.stringtemplate.v4.ST	getTemplate(java.lang.String name)

Methods inherited from class java.lang.Object

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

Methods inherited from interface io.mxnet.caffetranslator.SymbolGenerator

generate

Field Detail

gh

protected final GenerationHelper gh

Constructor Detail

BaseGenerator

public BaseGenerator()

Method Detail

getTemplate

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

generateVar

protected 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)