tensor.h File Reference

Auxiliary param for tensor primitive. More...

#include <dmlc/base.h>
#include <dmlc/parameter.h>
#include <nnvm/tuple.h>

Include dependency graph for tensor.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	nnvm::top::ConcatenateParam
struct	nnvm::top::ExpandDimsParam
struct	nnvm::top::SplitParam
struct	nnvm::top::TakeParam
struct	nnvm::top::StridedSliceParam
struct	nnvm::top::CastParam
struct	nnvm::top::IndicatorParam
struct	nnvm::top::ReshapeParam
struct	nnvm::top::SqueezeParam
struct	nnvm::top::ScalarParam
struct	nnvm::top::FillValueParam
struct	nnvm::top::TransposeParam
struct	nnvm::top::FlipParam
struct	nnvm::top::BroadcastToParam
struct	nnvm::top::ReduceParam
struct	nnvm::top::InitOpWithScalarParam
struct	nnvm::top::InitOpParam
struct	nnvm::top::ElementWiseReduceParam
struct	nnvm::top::MatMulParam
struct	nnvm::top::ClipParam
struct	nnvm::top::SliceLikeParam

Namespaces
	nnvm
	nnvm::top

Macros
#define	DMLC_DECLARE_DTYPE_FIELD(name)

Enumerations
enum	nnvm::top::TypeFlag {   nnvm::top::kFloat32 = 0, nnvm::top::kFloat64 = 1, nnvm::top::kFloat16 = 2, nnvm::top::kUint8 = 3,   nnvm::top::kInt32 = 4, nnvm::top::kInt8 = 5, nnvm::top::kInt64 = 6, nnvm::top::kInt16 = 7,   nnvm::top::kUint16 = 8, nnvm::top::kUint32 = 9, nnvm::top::kUint64 = 10 }
enum	nnvm::top::IndicatorRuleFlag { nnvm::top::kGT0 = 0, nnvm::top::kLT0 = 1, nnvm::top::kMax = 2, nnvm::top::kMin = 3 }

Detailed Description

Auxiliary param for tensor primitive.

Copyright (c) 2017 by Contributors

Macro Definition Documentation

#define DMLC_DECLARE_DTYPE_FIELD

(

name

)

Value:

DMLC_DECLARE_FIELD(name)                                          \
  .add_enum("float16", kFloat16)                                    \
  .add_enum("float32", kFloat32)                                    \
  .add_enum("float64", kFloat64)                                    \
  .add_enum("uint8",  kUint8)                                       \
  .add_enum("uint16", kUint16)                                      \
  .add_enum("uint32", kUint32)                                      \
  .add_enum("uint64", kUint64)                                      \
  .add_enum("int8",  kInt8)                                         \
  .add_enum("int16", kInt16)                                        \
  .add_enum("int32", kInt32)                                        \
  .add_enum("int64", kInt64)