下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

npu.keep_dtype_scope

功能说明

指定哪些算子保持原有精度,如果原始网络模型中的算子精度在昇腾AI处理器上不支持,则系统内部自动采用算子支持的高精度来计算。

函数原型

npu.keep_dtype_scope()

参数说明

无。

返回值

Python上下文管理器,在该上下文中的构图算子,会带有NPU识别的特殊属性。

调用示例

1
2
3
import npu_device as npu
with npu.keep_dtype_scope():
    v = tf.add(1, 1)
搜索结果
找到“0”个结果

当前产品无相关内容

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