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
200 node_to_context_.erase(node); in InferShapesForFunction()230 auto it = node_to_context_.find(input); in AddNodeInternal()231 if (it == node_to_context_.end()) { in AddNodeInternal()268 node_to_context_[node].swap(ec); in AddNodeInternal()307 auto it = node_to_context_.find(node); in UpdateNode()308 if (it == node_to_context_.end()) { in UpdateNode()327 auto iter = node_to_context_.find(input); in UpdateNode()328 if (iter == node_to_context_.end()) { in UpdateNode()
654 node_to_context_.reserve(graph.graph()->node_size()); in SymbolicShapeRefiner()760 auto it = node_to_context_.find(node); in GetNodeContext()761 if (it == node_to_context_.end()) { in GetNodeContext()768 auto it = node_to_context_.find(node); in GetContext()769 if (it == node_to_context_.end()) { in GetContext()1353 NodeContext& node_ctx = node_to_context_[node]; in AddNode()2045 absl::flat_hash_map<const NodeDef*, NodeContext> node_to_context_; member in tensorflow::grappler::SymbolicShapeRefiner