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.cc275 nodes_to_simplify->PushBack(node_to_idx_[fanout]); in OptimizeNode()
284 nodes_to_delete->insert(node_to_idx_[node]); in OptimizeNode()
316 nodes_to_simplify->PushBack(node_to_idx_[old_input_node]); in OptimizeNode()
322 nodes_to_simplify->PushBack(node_to_idx_[node]); in OptimizeNode()
435 nodes_to_simplify->PushBack(node_to_idx_[input]); in OptimizeNode()
443 nodes_to_simplify->PushBack(node_to_idx_[consumer]); in OptimizeNode()
553 const int input_node_idx = node_to_idx_[input_node]; in TransitiveReduction()
625 node_to_idx_.clear(); in BuildNodeToIdx()
628 node_to_idx_[&node] = i; in BuildNodeToIdx()
Ddependency_optimizer.h81 std::unordered_map<const NodeDef*, int> node_to_idx_; variable