Searched refs:node_to_context_ (Results 1 – 3 of 3) sorted by relevance
114 auto it = node_to_context_.find(node); in GetContext()115 if (it == node_to_context_.end()) { in GetContext()123 auto it = node_to_context_.find(node); in GetExtendedContext()124 if (it == node_to_context_.end()) { in GetExtendedContext()303 node_to_context_; variable
184 node_to_context_.erase(node); in InferShapesForFunction()214 auto it = node_to_context_.find(input); in AddNodeInternal()215 if (it == node_to_context_.end()) { in AddNodeInternal()252 node_to_context_[node].swap(ec); in AddNodeInternal()291 auto it = node_to_context_.find(node); in UpdateNode()292 if (it == node_to_context_.end()) { in UpdateNode()311 auto iter = node_to_context_.find(input); in UpdateNode()312 if (iter == node_to_context_.end()) { in UpdateNode()
625 node_to_context_.reserve(graph.graph()->node_size()); in SymbolicShapeRefiner()731 auto it = node_to_context_.find(node); in GetNodeContext()732 if (it == node_to_context_.end()) { in GetNodeContext()739 auto it = node_to_context_.find(node); in GetContext()740 if (it == node_to_context_.end()) { in GetContext()1315 NodeContext& node_ctx = node_to_context_[node]; in AddNode()2001 absl::flat_hash_map<const NodeDef*, NodeContext> node_to_context_; member in tensorflow::grappler::SymbolicShapeRefiner