io.mxnet.caffetranslator.generators

Class FlattenGenerator

java.lang.Object

io.mxnet.caffetranslator.generators.BaseGenerator

io.mxnet.caffetranslator.generators.FlattenGenerator

All Implemented Interfaces:

SymbolGenerator

public class FlattenGenerator
extends BaseGenerator

Field Summary

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

gh

Constructor Summary

Constructors

Constructor and Description
FlattenGenerator()

Method Summary

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

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

FlattenGenerator

public FlattenGenerator()

Method Detail

generate

public GeneratorOutput generate(Layer layer,
                                MLModel model)