文档
注册

简介

算子信息库作为算子开发的交付件之一,主要体现算子在昇腾AI处理器上的具体实现规格。针对TBE算子,算子信息库包括算子支持的输入输出dtype、format以及输入shape等信息;针对AI CPU算子,算子信息库包括算子输入输出的name,支持的dtype、format等信息。网络运行时,图编译器会根据算子信息库中的算子信息做基本校验,并进行算子匹配。

算子匹配的规则如下:

图编译器优先查找TBE算子信息库,若能匹配到TBE算子信息库中的算子,则选择TBE算子进行图编译;若无法匹配到TBE算子信息库中的算子,则查找AI CPU算子信息库,选择AI CPU算子进行图编译。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词