Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc253 nodes_to_simplify->PushBack(node_to_idx_[fanout]); in OptimizeNode()
262 nodes_to_delete->insert(node_to_idx_[node]); in OptimizeNode()
294 nodes_to_simplify->PushBack(node_to_idx_[old_input_node]); in OptimizeNode()
299 nodes_to_simplify->PushBack(node_to_idx_[node]); in OptimizeNode()
411 nodes_to_simplify->PushBack(node_to_idx_[input]); in OptimizeNode()
420 nodes_to_simplify->PushBack(node_to_idx_[consumer]); in OptimizeNode()
496 const int input_node_idx = node_to_idx_[input_node]; in TransitiveReduction()
582 node_to_idx_.clear(); in BuildNodeToIdx()
585 node_to_idx_[&node] = i; in BuildNodeToIdx()
Ddependency_optimizer.h78 std::unordered_map<const NodeDef*, int> node_to_idx_; variable