Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dloop_optimizer.cc77 std::map<NodeDef*, int> invariant_nodes_; member in tensorflow::grappler::__anon1ad005370111::LoopInvariantNodeMotionOptimizer
100 if (invariant_nodes_.count(consumer)) { in HandleInvariantEnter()
145 if (invariant_nodes_.count(consumer)) { in HandleConst()
217 if (!invariant_nodes_.count(consumer)) { in HandleInvariantNode()
261 for (auto iter = invariant_nodes_.begin(); iter != invariant_nodes_.end(); in MoveInvariantNodes()
279 for (auto iter = invariant_nodes_.begin(); iter != invariant_nodes_.end();) { in RevertInvariantNodes()
285 if (!invariant_nodes_.count(consumer)) { in RevertInvariantNodes()
289 invariant_nodes_.erase(iter++); in RevertInvariantNodes()
306 auto iter = invariant_nodes_.find(producer); in RevertInvariantNodes()
307 if (iter != invariant_nodes_.end()) { in RevertInvariantNodes()
[all …]