mxnet
|
Operator interface of mxnet. More...
#include <dmlc/base.h>
#include <dmlc/logging.h>
#include <dmlc/registry.h>
#include <mshadow/tensor.h>
#include <string>
#include <vector>
#include <utility>
#include "./base.h"
#include "./resource.h"
#include <mxnet/ndarray.h>
Go to the source code of this file.
Classes | |
class | mxnet::Optimizer |
struct | mxnet::OptimizerReg |
Registry entry for Optimizer factory functions. More... | |
Namespaces | |
mxnet | |
namespace of mxnet | |
Macros | |
#define | MXNET_REGISTER_OPTIMIZER(name, OptimizerType) |
Macro to register Optimizer. More... | |
Typedefs | |
typedef std::function < Optimizer *()> | mxnet::OptimizerFactory |
typedef the factory function of Optimizer More... | |
Operator interface of mxnet.
Copyright (c) 2015 by Contributors
#define MXNET_REGISTER_OPTIMIZER | ( | name, | |
OptimizerType | |||
) |
Macro to register Optimizer.