據(jù)類型API)
InstanceOutputDataType【免費下載鏈接】geGEGraph Engine是面向昇騰的圖編譯器和執(zhí)行器提供了計算圖優(yōu)化、多流并行、內(nèi)存復用和模型下沉等技術手段加速模型執(zhí)行效率減少模型內(nèi)存占用。 GE 提供對 PyTorch、TensorFlow 前端的友好接入能力并同時支持 onnx、pb 等主流模型格式的解析與編譯。項目地址: https://gitcode.com/cann/ge產(chǎn)品支持情況產(chǎn)品是否支持Atlas A3 訓練系列產(chǎn)品 / Atlas A3 推理系列產(chǎn)品√Atlas A2 訓練系列產(chǎn)品 / Atlas A2 推理系列產(chǎn)品√頭文件/庫文件頭文件#include ge/compliant_node_builder.h庫文件libeager_style_graph_builder_base.so、libeager_style_graph_builder_base_static.a功能說明設置輸出數(shù)據(jù)類型。函數(shù)原型CompliantNodeBuilder InstanceOutputDataType(const char_t *name, ge::DataType data_type);參數(shù)說明參數(shù)名輸入/輸出說明name輸入輸出名稱。data_type輸入數(shù)據(jù)類型。返回值說明參數(shù)名類型說明-CompliantNodeBuilder 當前構建器對象的引用支持鏈式調(diào)用。約束說明無【免費下載鏈接】geGEGraph Engine是面向昇騰的圖編譯器和執(zhí)行器提供了計算圖優(yōu)化、多流并行、內(nèi)存復用和模型下沉等技術手段加速模型執(zhí)行效率減少模型內(nèi)存占用。 GE 提供對 PyTorch、TensorFlow 前端的友好接入能力并同時支持 onnx、pb 等主流模型格式的解析與編譯。項目地址: https://gitcode.com/cann/ge創(chuàng)作聲明:本文部分內(nèi)容由AI輔助生成(AIGC),僅供參考