Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.h251 std::unordered_set<string> nodes_whitelist_; variable
Dconstant_folding.cc863 nodes_whitelist_.find(node.name()) == nodes_whitelist_.end()) { in IsFoldable()
929 nodes_whitelist_.find(node.name()) == nodes_whitelist_.end()) { in IsFoldable()
3372 nodes_whitelist_.clear(); in RunOptimizationPass()
3383 nodes_whitelist_.insert(fetch_node->name()); in RunOptimizationPass()