昇腾社区首页
中文
注册
开发者
下载

简介

C风格的图构建器结构体,C、C++的内部实现类,C用户构图调用esb_funcs.h接口即可,C++用户调用EsGraphBuilder类的对外接口即可。

需要包含的头文件

1
#include <es_c_graph_builder.h>

Public成员函数

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
EsCGraphBuilder()
explicit EsCGraphBuilder(const char *name)
~EsCGraphBuilder()
EsCGraphBuilder(EsCGraphBuilder &&) noexcept
EsCGraphBuilder &operator=(EsCGraphBuilder &&) noexcept
EsCTensorHolder *GetTensorHolderFromNode(const ge::GNode &node, int32_t output_index)
EsCTensorHolder *AppendGraphInput(const ge::char_t *name = nullptr, const ge::char_t *type = nullptr)
EsCTensorHolder *AddGraphInput(int32_t index, const ge::char_t *name = nullptr, const ge::char_t *type = nullptr, C_DataType data_type = C_DT_FLOAT, C_Format format = C_FORMAT_ND, const int64_t *dims = nullptr, int64_t dim_num = 0)
ge::Status SetGraphOutput(EsCTensorHolder *tensor, int64_t output_index)
ge::Graph *GetGraph()
std::unique_ptr<ge::Graph> BuildGraphAndReset()
ge::AscendString GenerateNodeName(const ge::char_t *node_type)
std::vector<EsCTensorHolder *> *CreateDynamicTensorHolderFromNode(const ge::GNode& node, int32_t start_idx, int32_t output_num)
T *AddResource(std::unique_ptr<T> value, std::function<void(void*)> deleter = DefaultDeleter<T>)