未验证 提交 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,
const std::string& mixed_params_file,
phi::DataType mixed_precision,
phi::Backend backend,
bool keep_io_types = true,
std::unordered_set<std::string> black_list = {});
bool keep_io_types,
const std::unordered_set<std::string>& black_list);
} // namespace analysis
} // namespace inference
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册