npu.keep_dtype_scope
函数原型
npu.keep_dtype_scope()
功能说明
指定哪些算子保持原有精度。
参数说明
无。
返回值
Python上下文管理器，在该上下文中的构图算子，会带有NPU识别的特殊属性。
调用示例
import npu_device as npu
with npu.keep_dtype_scope():
v = tf.add(1, 1)
父主题： TF Adapter 2.6 接口参考
npu.keep_dtype_scope()
指定哪些算子保持原有精度。
无。
Python上下文管理器，在该上下文中的构图算子，会带有NPU识别的特殊属性。
import npu_device as npu
with npu.keep_dtype_scope():
v = tf.add(1, 1)