Searched refs:node_to_context_ (Results 1 – 3 of 3) sorted by relevance
130 auto it = node_to_context_.find(node); in GetContext()131 if (it == node_to_context_.end()) { in GetContext()139 auto it = node_to_context_.find(node); in GetExtendedContext()140 if (it == node_to_context_.end()) { in GetExtendedContext()272 node_to_context_; variable
191 nested_inferences[node_name] = std::move(node_to_context_[node]); in InferShapesForFunction()192 node_to_context_.erase(node); in InferShapesForFunction()204 node_to_context_.erase(node); in InferShapesForFunction()223 auto it = node_to_context_.find(input); in AddNode()224 if (it == node_to_context_.end()) { in AddNode()274 node_to_context_[node].swap(ec); in AddNode()313 auto it = node_to_context_.find(node); in UpdateNode()314 if (it == node_to_context_.end()) { in UpdateNode()333 auto iter = node_to_context_.find(input); in UpdateNode()334 if (iter == node_to_context_.end()) { in UpdateNode()
557 node_to_context_.reserve(graph.graph()->node_size()); in SymbolicShapeRefiner()574 auto it = node_to_context_.find(node); in GetNodeContext()575 if (it == node_to_context_.end()) { in GetNodeContext()582 auto it = node_to_context_.find(node); in GetContext()583 if (it == node_to_context_.end()) { in GetContext()1071 NodeContext& node_ctx = node_to_context_[node]; in AddNode()1691 std::unordered_map<const NodeDef*, NodeContext> node_to_context_; member in tensorflow::grappler::SymbolicShapeRefiner