Home
last modified time | relevance | path

Searched refs:rewrite_rule (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgraph_optimizer_stage.h169 const string& rewrite_rule) const { in OptimizedNodeName() argument
170 const string prefix = strings::StrCat(stage_name_, "_", rewrite_rule); in OptimizedNodeName()
179 const string& rewrite_rule) { in UniqueOptimizedNodeName() argument
180 const string node_name = OptimizedNodeName(node, rewrite_rule); in UniqueOptimizedNodeName()
/external/tensorflow/tensorflow/core/graph/
Dmkl_layout_pass.cc610 std::function<bool(const Node*)> rewrite_rule; member
3247 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()