From 923b9ee66d9ed37c7a266669d291c6e8056e9333 Mon Sep 17 00:00:00 2001 From: Zhennan Qin Date: Thu, 20 Dec 2018 18:10:18 +0800 Subject: [PATCH] Fix build --- src/operator/quantization/quantize_graph_pass.cc | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/operator/quantization/quantize_graph_pass.cc b/src/operator/quantization/quantize_graph_pass.cc index 676615cf0402..af533978a6f5 100644 --- a/src/operator/quantization/quantize_graph_pass.cc +++ b/src/operator/quantization/quantize_graph_pass.cc @@ -64,7 +64,7 @@ NodePtr InsertNode(std::string op_name, } std::vector OfflineParams(std::vector&& outputs, - const std::unordered_set&& offline_params) { + const std::unordered_set& offline_params) { std::string node_suffixs[3] = {"", "_min", "_max"}; std::unordered_map mirror_map; nnvm::NodeEntryMap entry_var; @@ -89,7 +89,8 @@ std::vector OfflineParams(std::vector&& outputs, return outputs; } -inline bool NeedQuantize(NodePtr node, const std::unordered_set& excluded_nodes) { +inline bool NeedQuantize(const NodePtr node, + const std::unordered_set& excluded_nodes) { static auto& quantized_op_map = Op::GetAttr("FQuantizedOp"); static auto& fexec_type = nnvm::Op::GetAttr("FExecType"); const auto& op = node->op();