Searched refs:rewrite_rule (Results 1 – 2 of 2) sorted by relevance
169 const string& rewrite_rule) const { in OptimizedNodeName() argument170 const string prefix = strings::StrCat(stage_name_, "_", rewrite_rule); in OptimizedNodeName()179 const string& rewrite_rule) { in UniqueOptimizedNodeName() argument180 const string node_name = OptimizedNodeName(node, rewrite_rule); in UniqueOptimizedNodeName()
610 std::function<bool(const Node*)> rewrite_rule; member3247 if (n->type_string().compare(ri->name) == 0 && ri->rewrite_rule(n)) { in CheckForQuantizedNodeRewrite()3330 if (n->type_string().compare(ri->name) == 0 && ri->rewrite_rule(n)) { in CheckForNodeRewrite()