據(jù)集緩沖區(qū)API)
aclmdlGetDatasetBuffer【免費(fèi)下載鏈接】geGEGraph Engine是面向昇騰的圖編譯器和執(zhí)行器提供了計(jì)算圖優(yōu)化、多流并行、內(nèi)存復(fù)用和模型下沉等技術(shù)手段加速模型執(zhí)行效率減少模型內(nèi)存占用。 GE 提供對(duì) PyTorch、TensorFlow 前端的友好接入能力并同時(shí)支持 onnx、pb 等主流模型格式的解析與編譯。項(xiàng)目地址: https://gitcode.com/cann/ge產(chǎn)品支持情況產(chǎn)品是否支持Atlas A3 訓(xùn)練系列產(chǎn)品 / Atlas A3 推理系列產(chǎn)品√Atlas A2 訓(xùn)練系列產(chǎn)品 / Atlas A2 推理系列產(chǎn)品√功能說(shuō)明獲取aclmdlDataset中的第n個(gè)aclDataBuffer。函數(shù)原型aclDataBuffer* aclmdlGetDatasetBuffer(const aclmdlDataset *dataset, size_t index)參數(shù)說(shuō)明參數(shù)名輸入/輸出說(shuō)明dataset輸入aclmdlDataset類型的指針。需提前調(diào)用aclmdlCreateDataset接口創(chuàng)建aclmdlDataset類型的數(shù)據(jù)。index輸入表明獲取的是第幾個(gè)aclDataBuffer。返回值說(shuō)明獲取成功返回aclDataBuffer的地址。獲取失敗返回空地址。【免費(fèi)下載鏈接】geGEGraph Engine是面向昇騰的圖編譯器和執(zhí)行器提供了計(jì)算圖優(yōu)化、多流并行、內(nèi)存復(fù)用和模型下沉等技術(shù)手段加速模型執(zhí)行效率減少模型內(nèi)存占用。 GE 提供對(duì) PyTorch、TensorFlow 前端的友好接入能力并同時(shí)支持 onnx、pb 等主流模型格式的解析與編譯。項(xiàng)目地址: https://gitcode.com/cann/ge創(chuàng)作聲明:本文部分內(nèi)容由AI輔助生成(AIGC),僅供參考