文档
注册
评分
提单
论坛
小AI

keep_dtype_scope

功能说明

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

函数原型

def keep_dtype_scope()

参数说明

无。

使用约束

算子精度模式为保持原图精度(即precision_mode指定为must_keep_origin_dtype)时,该接口不生效。

返回值

无。

调用示例

1
2
with npu_scope.keep_dtype_scope(): 
    X = tf.conv2d(a)
搜索结果
找到“0”个结果

当前产品无相关内容

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