mxnet
|
Auxiliary param for tensor primitive. More...
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 } |
Auxiliary param for tensor primitive.
Copyright (c) 2017 by Contributors
#define DMLC_DECLARE_DTYPE_FIELD | ( | name | ) |