Operator
【免费下载链接】geGE(Graph Engine)是面向昇腾的图编译器和执行器,提供了计算图优化、多流并行、内存复用和模型下沉等技术手段,加速模型执行效率,减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力,并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge
简介
Operator构造函数和析构函数
AddControlInput
AttrRegister
BreakConnect
DynamicInputRegister
DynamicInputRegisterByIndex
DynamicOutputRegister
IsEmpty
InferShapeAndType
GetAttr
GetAllAttrNamesAndTypes
GetAllIrAttrNamesAndTypes
GetDynamicInputNum
GetDynamicInputDesc
GetDynamicOutputNum
GetDynamicOutputDesc
GetDynamicSubgraph
GetDynamicSubgraphBuilder
GetInferenceContext
GetInputAttr
GetInputConstData
GetInputsSize
GetInputDesc
GetName
GetSubgraph
GetSubgraphBuilder
GetSubgraphNamesCount
GetSubgraphNames
GetOpType
GetOutputAttr
GetOutputDesc
GetOutputsSize
SetAttr
SetInput
SetInferenceContext
SetInputAttr
SetOutputAttr
SetSubgraphBuilder
SetSubgraphInstanceName
SubgraphCountRegister
TryGetInputDesc
UpdateInputDesc
UpdateOutputDesc
UpdateDynamicInputDesc
UpdateDynamicOutputDesc
VerifyAllAttr
【免费下载链接】geGE(Graph Engine)是面向昇腾的图编译器和执行器,提供了计算图优化、多流并行、内存复用和模型下沉等技术手段,加速模型执行效率,减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力,并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考