add the overload quantize_int8 function api to the quantization head file.

11a0f1a9 · Shucai Xiao · a8fadc6f · 11a0f1a9
Commit 11a0f1a9 authored Jun 05, 2019 by Shucai Xiao
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

src/include/migraphx/quantization.hpp src/include/migraphx/quantization.hpp +2 -0

No files found.
--- a/src/include/migraphx/quantization.hpp
+++ b/src/include/migraphx/quantization.hpp
@@ -15,6 +15,8 @@ struct program;
 void quantize(program& prog, const std::vector<std::string>& ins_names);
 void quantize(program& prog);

+void quantize_int8(program& prog);
+void quantize_int8(program& prog, const std::vector<std::string>& ins_names);
 void quantize_int8(program& prog,
                   const std::vector<std::string>& ins_names,
                   const std::vector<std::pair<float, float>>& quant_params);