點API)
RemoveNode【免費下載鏈接】geGEGraph Engine是面向昇騰的圖編譯器和執(zhí)行器提供了計算圖優(yōu)化、多流并行、內存復用和模型下沉等技術手段加速模型執(zhí)行效率減少模型內存占用。 GE 提供對 PyTorch、TensorFlow 前端的友好接入能力并同時支持 onnx、pb 等主流模型格式的解析與編譯。項目地址: https://gitcode.com/cann/ge產品支持情況產品是否支持Atlas A3 訓練系列產品 / Atlas A3 推理系列產品√Atlas A2 訓練系列產品 / Atlas A2 推理系列產品√頭文件/庫文件頭文件#include graph/graph.h庫文件libgraph.so功能說明刪除圖中的指定節(jié)點并刪除節(jié)點之間的連邊。函數(shù)原型graphStatus RemoveNode(GNode node) graphStatus RemoveNode(GNode node, bool contain_subgraph)參數(shù)說明參數(shù)名輸入/輸出說明node輸入刪除圖中的指定節(jié)點如果是子圖中的節(jié)點則需要指定contain_subgraph為true。contain_subgraph輸入刪除的指定節(jié)點是否在子圖中。返回值說明參數(shù)名類型說明-graphStatusGRAPH_SUCCESS(0)成功。其他值失敗。約束說明無【免費下載鏈接】geGEGraph Engine是面向昇騰的圖編譯器和執(zhí)行器提供了計算圖優(yōu)化、多流并行、內存復用和模型下沉等技術手段加速模型執(zhí)行效率減少模型內存占用。 GE 提供對 PyTorch、TensorFlow 前端的友好接入能力并同時支持 onnx、pb 等主流模型格式的解析與編譯。項目地址: https://gitcode.com/cann/ge創(chuàng)作聲明:本文部分內容由AI輔助生成(AIGC),僅供參考