Dragon - C++ API
A Computation Graph Virtual Machine Based Deep Learning Framework
Classes | Namespaces | Macros | Typedefs | Functions
dragon.h File Reference
#include <string>
#include <cstdint>
#include <vector>

Go to the source code of this file.

Classes

class  dragon::Device
 

Namespaces

 dragon
 

Macros

#define DRAGON_API
 

Typedefs

typedef struct float16 * dragon::float16_t
 
typedef struct GraphDef * dragon::GraphDef_t
 
typedef struct Workspace * dragon::Workspace_t
 

Functions

Workspace * dragon::CreateWorkspace (const std::string &name)
 
Workspace * dragon::ResetWorkspace (Workspace_t ws)
 
Workspace * dragon::ResetWorkspace (const std::string &name)
 
void dragon::MoveWorkspace (Workspace_t dst, Workspace_t src)
 
void dragon::DestroyWorkspace (Workspace_t ws)
 
void dragon::DestroyWorkspace (const std::string &name)
 
std::string dragon::CreateGraph (const GraphDef_t graph_def, const Device &device, Workspace_t ws)
 
std::string dragon::CreateGraph (const std::string &graph_file, const Device &device, Workspace_t ws)
 
void dragon::RunGraph (const std::string &graph_name, Workspace_t ws, const int stream_id)
 
void dragon::CreateTensor (const std::string &name, Workspace_t ws)
 
template<typename T >
DRAGON_API T * dragon::FetchTensor (const std::string &name, std::vector< int64_t > &shape, Workspace_t ws, const bool copy=false)
 
template<typename T >
DRAGON_API void dragon::FeedTensor (const std::string &name, const std::vector< int64_t > &shape, const T *data, const Device &device, Workspace_t ws)
 
DRAGON_API void dragon::CreateGraphDef (GraphDef_t *graph_def)
 
DRAGON_API void dragon::DestroyGraphDef (GraphDef_t graph_def)
 
void dragon::LoadCaffeModel (const std::string &model_file, Workspace_t ws)
 
void dragon::LoadONNXModel (const std::string &model_file, GraphDef_t init_graph, GraphDef_t pred_graph, std::vector< std::string > &inputs, std::vector< std::string > &outputs)
 
void dragon::SetLoggingLevel (const std::string &level)
 

Macro Definition Documentation

◆ DRAGON_API

#define DRAGON_API

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>