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.


class  dragon::Device




#define DRAGON_API


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


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


#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,