nn.h File Reference

Auxiliary param for tensor primitive. More...

#include <dmlc/base.h>
#include <dmlc/parameter.h>
#include <nnvm/tuple.h>
#include <nnvm/layout.h>
#include <string>
#include "tensor.h"

Include dependency graph for nn.h:

Go to the source code of this file.

Classes
struct	nnvm::top::DenseParam
struct	nnvm::top::DropoutParam
struct	nnvm::top::BatchNormParam
struct	nnvm::top::SoftmaxParam
struct	nnvm::top::LeakyReLUParam
struct	nnvm::top::PReLUParam
struct	nnvm::top::PadParam
struct	nnvm::top::Conv2DParam
struct	nnvm::top::WinogradWeightTransformParam
struct	nnvm::top::WinogradNNPACKWeightTransformParam
struct	nnvm::top::WinogradConv2DParam
struct	nnvm::top::Conv2DTransposeParam
struct	nnvm::top::MaxPool2DParam
struct	nnvm::top::AvgPool2DParam
struct	nnvm::top::GlobalPool2DParam
struct	nnvm::top::UpSamplingParam
struct	nnvm::top::LayoutTransformParam
struct	nnvm::top::MultiBoxPriorParam
struct	nnvm::top::MultiBoxTransformLocParam
struct	nnvm::top::NonMaximumSuppressionParam
struct	nnvm::top::LRNParam
struct	nnvm::top::L2NormalizeParam

Namespaces
	nnvm
	nnvm::top

Detailed Description

Auxiliary param for tensor primitive.

Copyright (c) 2017 by Contributors