未验证 提交 0972d6ac 编写于 作者: Y Yuanle Liu 提交者: GitHub

[Paddle Inference] improve convert_to_mixed_precision (#47333)

上级 5429d145
...@@ -51,8 +51,8 @@ void ConvertToMixedPrecision(const std::string& model_file, ...@@ -51,8 +51,8 @@ void ConvertToMixedPrecision(const std::string& model_file,
const std::string& mixed_params_file, const std::string& mixed_params_file,
phi::DataType mixed_precision, phi::DataType mixed_precision,
phi::Backend backend, phi::Backend backend,
bool keep_io_types = true, bool keep_io_types,
std::unordered_set<std::string> black_list = {}); const std::unordered_set<std::string>& black_list);
} // namespace analysis } // namespace analysis
} // namespace inference } // namespace inference
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册