def keep_dtype_scope()
指定哪些算子保持原有精度。
算子精度模式为保持原图精度(即precision_mode指定为must_keep_origin_dtype)时,该接口不生效。
无。
with npu_scope.keep_dtype_scope(): X = tf.conv2d(a)