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

接口列表

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

您可以在CANN软件安装后文件存储路径>“include”路径查看对应接口的头文件。

接口分类

头文件路径/库文件

接口列表

Parser解析接口

  • 头文件
    • /include/parser/tensorflow_parser.h
    • /include/parser/caffe_parser.h

    库文件:libfmk_parser.so

  • 头文件:/include/parser/onnx_parser.h

    库文件:libfmk_onnx_parser.so

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

aclgrphParseTensorFlow

aclgrphParseCaffe

aclgrphParseONNX

aclgrphParseONNXFromMem

Graph构建接口

  • 头文件:/include/graph/graph.h
  • 库文件:libgraph.so

Graph构造函数和析构函数

SetInputs

SetOutputs

SetTargets

IsValid

SetNeedIteration

AddOp

FindOpByName

GetAllOpName

SaveToFile

LoadFromFile

FindOpByType

GetName

CopyFrom

LoadFromMem

SaveToMem

LoadFromSerializedModelArray

Graph修改接口>

Graph类

  • 头文件:/include/graph/graph.h
  • 库文件:libgraph.so

GetAllNodes

GetDirectNode

RemoveNode

RemoveEdge

AddNodeByOp

AddDataEdge

AddControlEdge

ConstructFromInputs

Graph修改接口>

GNode类

  • 头文件:/include/graph/gnode.h
  • 库文件:libgraph.so

GNode构造函数和析构函数

GetType

GetName

GetInDataNodesAndPortIndexs

GetInControlNodes

GetOutDataNodesAndPortIndexs

GetOutControlNodes

GetInputConstData

GetInputIndexByName

GetDynamicInputIndexesByName

GetOutputIndexByName

GetDynamicOutputIndexesByName

GetInputsSize

GetOutputsSize

GetInputDesc

UpdateInputDesc

GetOutputDesc

UpdateOutputDesc

SetAttr

GetAttr

HasAttr

GetSubgraph

GetALLSubgraphs

Graph修改接口>CustomPassContext类

  • 头文件:/include/register/register_custom_pass.h
  • 库文件:libregister.so

CustomPassContext构造函数和析构函数

SetErrorMessage

Graph修改接口>REGISTER_CUSTOM_PASS

  • 头文件:/include/register/register_custom_pass.h
  • 库文件:libregister.so

REGISTER_CUSTOM_PASS

Graph修改接口>PassRegistrationData类

  • 头文件:/include/register/register_custom_pass.h
  • 库文件:libregister.so

PassRegistrationData构造函数和析构函数

CustomPassFn

Graph修改接口>PassReceiver类

  • 头文件:/include/register/register_custom_pass.h
  • 库文件:libregister.so

PassReceiver构造函数和析构函数

Graph修改接口>

Session类

  • 头文件:/include/ge/ge_api.h
  • 库文件:libge_runner.so

ShardGraphs

ShardGraphsToFile

SaveGraphsToPb

Graph修改接口>

GraphBuffer类

  • 头文件:/include/graph/graph_buffer.h
  • 库文件:libgraph_base.so

GraphBuffer的构造函数和析构函数

GetData

GetSize

Graph编译接口

  • 头文件:/include/ge/ge_ir_build.h
  • 库文件:libge_compiler.so

aclgrphBuildInitialize

aclgrphBuildModel

aclgrphSaveModel

aclgrphBuildFinalize

aclgrphConvertToWeightRefreshableGraphs

aclgrphBundleBuildModel

aclgrphBundleSaveModel

Graph运行接口

  • 头文件:/include/ge/ge_api.h
  • 库文件:libge_runner.so
  • 头文件:/include/ge/ge_graph_compile_summary.h
  • 库文件:libge_compiler.so

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

GEInitialize

GEFinalize

Session构造函数和析构函数

GetVariables

GetSessionId

AddGraph

AddGraphWithCopy

RemoveGraph

LoadGraph

GeSessionLoadGraph

RunGraph

RunGraphAsync

RunGraphWithStreamAsync

ExecuteGraphWithStreamAsync

GeSessionExecuteGraphWithStreamAsync

RunGraphDistribute

BuildGraph

CompileGraph

RegisterCallBackFunc

IsGraphNeedRebuild

GetCompiledGraphSummary

SetGraphConstMemoryBase

UpdateGraphFeatureMemoryBase

SetGraphFixedFeatureMemoryBase

SetGraphFixedFeatureMemoryBaseWithType

UpdateGraphRefreshableFeatureMemoryBase

RegisterExternalAllocator

UnregisterExternalAllocator

PaRemapped

GEGetErrorMsg

GEGetWarningMsg

GEGetErrorMsgV2

GEGetWarningMsgV2

Graph查询接口

  • 头文件:/include/ge/ge_api.h
  • 库文件:libge_runner.so

GetModelDistributeDesc

量化接口

  • 头文件:/include/amct/acl_graph_calibration.h
  • 库文件:libamctacl.so

aclgrphCalibration

维测接口

  • 头文件:/include/ge/ge_prof.h
  • 库文件:libmsprofiler.so

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

aclgrphProfInit

aclgrphProfFinalize

aclgrphProfCreateConfig

aclgrphProfDestroyConfig

aclgrphProfStart

aclgrphProfStop

其他接口

  • 头文件:/include/ge/ge_ir_build.h
  • 库文件:libge_compiler.so
  • 头文件:/include/ge/ge_api_error_codes.h
  • 库文件:libge_common_base.so

aclgrphDumpGraph

aclgrphSetOpAttr

aclgrphGetIRVersion

aclgrphGenerateForOp

GE_ERRORNO

数据类型

  • 头文件:/include/ge/ge_ir_build.h
  • 库文件:libge_compiler.so

ModelBufferData

/include/ge/ge_api_types.h

aclgrphBuildInitialize支持的配置参数

aclgrphBuildModel支持的配置参数

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

InputTensorInfo

OutputTensorInfo

MemoryType

  • 头文件:/include/ge/ge_prof.h
  • 库文件:libmsprofiler.so

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

ProfDataTypeConfig

ProfilingAicoreMetrics

/include/ge/ge_api_types.h

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

options参数说明

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

当前产品无相关内容

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