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)