Dragon - C++ API
A Computation Graph Virtual Machine Based Deep Learning Framework
Classes | Namespaces | Macros | Functions
py_dragon.h File Reference
#include "py_types.h"
#include "core/common.h"
#include "core/context.h"
#include "core/operator.h"
#include "core/registry.h"
#include "core/workspace.h"
#include "core/context_cuda.h"
#include "core/graph_gradient.h"
#include "core/operator_gradient.h"
#include "utils/caffemodel.h"
#include "onnx/onnx_backend.h"
#include <pybind11/stl.h>
#include <pybind11/pybind11.h>

Go to the source code of this file.

Classes

class  dragon::python::TensorFetcherBase
 
class  dragon::python::TensorFeederBase
 
class  dragon::python::NumpyFetcher
 
class  dragon::python::StringFetcher
 
class  dragon::python::NumpyFeeder
 

Namespaces

 dragon
 
 dragon::python
 

Macros

#define DRAGON_PYTHON_PY_DRAGON_H_
 
#define NPY_NO_DEPRECATED_API   NPY_1_7_API_VERSION
 
#define REGISTER_TENSOR_FETCHER(type, ...)   REGISTER_TYPED_CLASS(TensorFetcherRegistry, type, __VA_ARGS__)
 
#define REGISTER_TENSOR_FEEDER(type, ...)   REGISTER_TYPED_CLASS(TensorFeederRegistry, type, __VA_ARGS__)
 
#define NPY_NO_DEPRECATED_API   NPY_1_7_API_VERSION
 
#define REGISTER_TENSOR_FETCHER(type, ...)   REGISTER_TYPED_CLASS(TensorFetcherRegistry, type, __VA_ARGS__)
 
#define REGISTER_TENSOR_FEEDER(type, ...)   REGISTER_TYPED_CLASS(TensorFeederRegistry, type, __VA_ARGS__)
 

Functions

 dragon::python::DECLARE_TYPED_REGISTRY (TensorFetcherRegistry, TypeId, TensorFetcherBase)
 
TensorFetcherBase * dragon::python::CreateFetcher (TypeId type)
 
 dragon::python::DECLARE_TYPED_REGISTRY (TensorFeederRegistry, TypeId, TensorFeederBase)
 

Macro Definition Documentation

◆ DRAGON_PYTHON_PY_DRAGON_H_

#define DRAGON_PYTHON_PY_DRAGON_H_

◆ NPY_NO_DEPRECATED_API [1/2]

#define NPY_NO_DEPRECATED_API   NPY_1_7_API_VERSION

◆ NPY_NO_DEPRECATED_API [2/2]

#define NPY_NO_DEPRECATED_API   NPY_1_7_API_VERSION

Copyright (c) 2017-present, SeetaTech, Co.,Ltd.

Licensed under the BSD 2-Clause License. You should have received a copy of the BSD 2-Clause License along with the software. If not, See,

 <https://opensource.org/licenses/BSD-2-Clause>

◆ REGISTER_TENSOR_FEEDER [1/2]

#define REGISTER_TENSOR_FEEDER (   type,
  ... 
)    REGISTER_TYPED_CLASS(TensorFeederRegistry, type, __VA_ARGS__)

◆ REGISTER_TENSOR_FEEDER [2/2]

#define REGISTER_TENSOR_FEEDER (   type,
  ... 
)    REGISTER_TYPED_CLASS(TensorFeederRegistry, type, __VA_ARGS__)

◆ REGISTER_TENSOR_FETCHER [1/2]

#define REGISTER_TENSOR_FETCHER (   type,
  ... 
)    REGISTER_TYPED_CLASS(TensorFetcherRegistry, type, __VA_ARGS__)

◆ REGISTER_TENSOR_FETCHER [2/2]

#define REGISTER_TENSOR_FETCHER (   type,
  ... 
)    REGISTER_TYPED_CLASS(TensorFetcherRegistry, type, __VA_ARGS__)