Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Fix build
Browse files Browse the repository at this point in the history
  • Loading branch information
ZhennanQin committed Dec 20, 2018
1 parent f3a2dd2 commit e3287cd
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/operator/quantization/quantize_graph_pass.cc
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,7 @@ NodePtr InsertNode(std::string op_name,
}

std::vector<NodeEntry> OfflineParams(std::vector<NodeEntry>&& outputs,
const std::unordered_set<std::string>&& offline_params) {
const std::unordered_set<std::string>& offline_params) {
std::string node_suffixs[3] = {"", "_min", "_max"};
std::unordered_map<Node*, NodePtr> mirror_map;
nnvm::NodeEntryMap<NodePtr> entry_var;
Expand All @@ -89,7 +89,7 @@ std::vector<NodeEntry> OfflineParams(std::vector<NodeEntry>&& outputs,
return outputs;
}

inline bool NeedQuantize(NodePtr node, const std::unordered_set<std::string>& excluded_nodes) {
inline bool NeedQuantize(const NodePtr node, const std::unordered_set<std::string>& excluded_nodes) {
static auto& quantized_op_map = Op::GetAttr<mxnet::FQuantizedOp>("FQuantizedOp");
static auto& fexec_type = nnvm::Op::GetAttr<FExecType>("FExecType");
const auto& op = node->op();
Expand Down

0 comments on commit e3287cd

Please sign in to comment.