Searched refs:GetNodeContext (Results 1 – 1 of 1) sorted by relevance
573 NodeContext* GetNodeContext(const NodeDef* node) { in GetNodeContext() function in tensorflow::grappler::SymbolicShapeRefiner658 auto* ctx = GetNodeContext(function_node); in UpdateFunction()746 NodeContext* ctx = GetNodeContext(node); in UpdateNode()749 ctx = CHECK_NOTNULL(GetNodeContext(node)); in UpdateNode()766 NodeContext* src_ctx = GetNodeContext(src); in UpdateNode()1947 auto* ctx = shape_refiner->GetNodeContext(queue_node); in UpdateQueue()1950 ctx = CHECK_NOTNULL(shape_refiner->GetNodeContext(queue_node)); in UpdateQueue()1995 auto ctx = shape_refiner->GetNodeContext(enqueue_node); in UpdateEnqueue()1998 ctx = CHECK_NOTNULL(shape_refiner->GetNodeContext(enqueue_node)); in UpdateEnqueue()2177 auto ctx = refiner.GetNodeContext(&node); in InferStatically()