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

keep_dtype_scope

函数原型

def keep_dtype_scope()

功能说明

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

使用约束

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

参数说明

无。

返回值

无。

调用示例

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

当前产品无相关内容

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