io.mxnet.caffetranslator.generators

Class PluginIntLayerGenerator

java.lang.Object

io.mxnet.caffetranslator.generators.BaseGenerator

io.mxnet.caffetranslator.generators.PluginIntLayerGenerator

All Implemented Interfaces:

SymbolGenerator

public class PluginIntLayerGenerator
extends BaseGenerator

Field Summary

Fields inherited from class io.mxnet.caffetranslator.generators.BaseGenerator

gh

Constructor Summary

Constructors

Constructor and Description
PluginIntLayerGenerator()

Method Summary

Modifier and Type	Method and Description
GeneratorOutput	generate(Layer layer, MLModel model)
GeneratorOutput	generate(Layer layer, MLModel model, int num_weight)

Methods inherited from class io.mxnet.caffetranslator.generators.BaseGenerator

generateVar, getTemplate

Methods inherited from class java.lang.Object

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

Constructor Detail

PluginIntLayerGenerator

public PluginIntLayerGenerator()

Method Detail

generate

public GeneratorOutput generate(Layer layer,
                                MLModel model)

generate

public GeneratorOutput generate(Layer layer,
                                MLModel model,
                                int num_weight)