简介
概述
TBE的Register模块,提供了注册管理算子compute函数、算子编译选项、算子编译阶段融合pass等能力的接口。
您可以在CANN软件安装后文件存储路径下的“python/site-packages/tbe/common/register/register_api.py”文件中找到Register模块提供的接口定义。
父主题: TBE Register API
TBE的Register模块,提供了注册管理算子compute函数、算子编译选项、算子编译阶段融合pass等能力的接口。
您可以在CANN软件安装后文件存储路径下的“python/site-packages/tbe/common/register/register_api.py”文件中找到Register模块提供的接口定义。
当前产品无相关内容
未找到相关内容,请尝试其他搜索词