Searched refs:node_to_context_ (Results 1 – 3 of 3) sorted by relevance
113 auto it = node_to_context_.find(node); in GetContext()114 if (it == node_to_context_.end()) { in GetContext()122 auto it = node_to_context_.find(node); in GetExtendedContext()123 if (it == node_to_context_.end()) { in GetExtendedContext()240 node_to_context_; variable
184 node_to_context_.erase(node); in InferShapesForFunction()209 auto it = node_to_context_.find(input); in AddNode()210 if (it == node_to_context_.end()) { in AddNode()247 node_to_context_[node].swap(ec); in AddNode()286 auto it = node_to_context_.find(node); in UpdateNode()287 if (it == node_to_context_.end()) { in UpdateNode()306 auto iter = node_to_context_.find(input); in UpdateNode()307 if (iter == node_to_context_.end()) { in UpdateNode()
559 node_to_context_.reserve(graph.graph()->node_size()); in SymbolicShapeRefiner()579 auto it = node_to_context_.find(node); in GetNodeContext()580 if (it == node_to_context_.end()) { in GetNodeContext()587 auto it = node_to_context_.find(node); in GetContext()588 if (it == node_to_context_.end()) { in GetContext()1151 NodeContext& node_ctx = node_to_context_[node]; in AddNode()1807 std::unordered_map<const NodeDef*, NodeContext> node_to_context_; member in tensorflow::grappler::SymbolicShapeRefiner