下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

简介

本文档主要描述给应用用户或者上层框架提供了安全易用的构图接口集合,开发者可以调用这些接口构建网络模型,设置模型所包含的图、图内的算子以及算子的属性信息(包括Input、Output及其他属性信息)。

您可以在CANN软件包安装路径查看对应接口的头文件。

表1

接口分类

头文件路径

备注

接口列表

Parser解析接口

/include/parser/tensorflow_parser.h

/include/parser/caffe_parser.h

/include/parser/onnx_parser.h

-

aclgrphParseTensorFlow

Atlas A2训练系列产品/Atlas 800I A2推理产品不支持Caffe框架

aclgrphParseCaffe

-

aclgrphParseONNX

-

aclgrphParseONNXFromMem

Graph构建接口

/include/graph/graph.h

-

Graph构造函数和析构函数

SetInputs

SetOutputs

SetTargets

IsValid

SetNeedIteration

AddOp

FindOpByName

GetAllOpName

SaveToFile

LoadFromFile

FindOpByType

GetName

CopyFrom

LoadFromMem

SaveToMem

LoadFromSerializedModelArray

Graph修改接口>

Graph类

/include/graph/graph.h

-

GetAllNodes

GetDirectNode

RemoveNode

RemoveEdge

AddNodeByOp

AddDataEdge

AddControlEdge

ConstructFromInputs

Graph修改接口>

GNode类

/include/graph/gnode.h

-

GNode构造函数和析构函数

GetType

GetName

GetInDataNodesAndPortIndexs

GetInControlNodes

GetOutDataNodesAndPortIndexs

GetOutControlNodes

GetInputConstData

GetInputIndexByName

GetOutputIndexByName

GetInputsSize

GetOutputsSize

GetInputDesc

UpdateInputDesc

GetOutputDesc

UpdateOutputDesc

SetAttr

GetAttr

HasAttr

GetSubgraph

GetALLSubgraphs

Graph修改接口>

Session类

/include/ge/ge_api.h

-

ShardGraphs

ShardGraphsToFile

SaveGraphsToPb

Graph修改接口>

GraphBuffer类

/include/graph/graph_buffer.h

-

GraphBuffer的构造函数和析构函数

GetData

GetSize

Graph编译接口

/include/ge/ge_ir_build.h

-

aclgrphBuildInitialize

aclgrphBuildModel

aclgrphSaveModel

aclgrphBuildFinalize

Graph运行接口

/include/ge/ge_api.h

/include/ge/ge_graph_compile_summary.h

Atlas 200/300/500 推理产品Atlas 200I/500 A2推理产品不支持此头文件中的接口

-

GEInitialize

GEFinalize

Session构造函数和析构函数

GetVariables

GetSessionId

AddGraph

AddGraphWithCopy

RemoveGraph

LoadGraph

RunGraph

RunGraphAsync

RunGraphWithStreamAsync

RunGraphDistribute

BuildGraph

CompileGraph

RegisterCallBackFunc

IsGraphNeedRebuild

GetCompiledGraphSummary

SetGraphConstMemoryBase

UpdateGraphFeatureMemoryBase

SetGraphFixedFeatureMemoryBase

SetGraphFixedFeatureMemoryBaseWithType

UpdateGraphRefreshableFeatureMemoryBase

RegisterExternalAllocator

UnregisterExternalAllocator

PaRemapped

GEGetErrorMsg

GEGetWarningMsg

GEGetErrorMsgV2

GEGetWarningMsgV2

Graph查询接口

/include/ge/ge_api.h

-

GetModelDistributeDesc

量化接口

/include/amct/acl_graph_calibration.h

-

aclgrphCalibration

维测接口

/include/ge/ge_prof.h

Atlas 200/300/500 推理产品Atlas 200I/500 A2推理产品不支持此头文件中的接口

-

aclgrphProfInit

aclgrphProfFinalize

aclgrphProfCreateConfig

aclgrphProfDestroyConfig

aclgrphProfStart

aclgrphProfStop

其他接口

/include/ge/ge_ir_build.h

/include/ge/ge_api_error_codes.h

-

aclgrphDumpGraph

aclgrphSetOpAttr

aclgrphGetIRVersion

aclgrphGenerateForOp

GE_ERRORNO

数据类型

/include/ge/ge_ir_build.h

-

ModelBufferData

/include/ge/ge_api_types.h

-

aclgrphBuildInitialize支持的配置参数

-

aclgrphBuildModel支持的配置参数

-

Parser解析接口支持的配置参数

-

InputTensorInfo

-

OutputTensorInfo

-

MemoryType

/include/ge/ge_prof.h

Atlas 200/300/500 推理产品Atlas 200I/500 A2推理产品不支持此头文件中的接口

-

ProfDataTypeConfig

ProfilingAicoreMetrics

/include/ge/ge_api_types.h

Atlas 200/300/500 推理产品Atlas 200I/500 A2推理产品不支持此头文件中的接口

-

options参数说明

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词