CANN/ge GESession API文档

GESession

【免费下载链接】geGE（Graph Engine）是面向昇腾的图编译器和执行器，提供了计算图优化、多流并行、内存复用和模型下沉等技术手段，加速模型执行效率，减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力，并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge

• 简介

• GeSession构造函数和析构函数

• AddGraph

• AddGraphClone

• CompileGraph

• GEFinalizeV2

• GEGetErrorMsgV3

• GEGetWarningMsgV3

• GEInitializeV2

• GetCompiledGraphSummary

• GetCompiledModel

• GetSessionId

• IsGraphNeedRebuild

• LoadGraph

• RegisterCallBackFunc

• RegisterExternalAllocator

• RemoveGraph

• RunGraph

• RunGraphAsync

• RunGraphWithStreamAsync

• SetGraphConstMemoryBase

• SetGraphFixedFeatureMemoryBaseWithType

• UpdateGraphFeatureMemoryBase

• UpdateGraphRefreshableFeatureMemoryBase

• UnregisterExternalAllocator

【免费下载链接】geGE（Graph Engine）是面向昇腾的图编译器和执行器，提供了计算图优化、多流并行、内存复用和模型下沉等技术手段，加速模型执行效率，减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力，并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge