Searched refs:GetNodeContext (Results 1 – 1 of 1) sorted by relevance
730 NodeContext* GetNodeContext(const NodeDef* node) { in GetNodeContext() function in tensorflow::grappler::SymbolicShapeRefiner772 auto* ctx = GetNodeContext(function_node); in UpdateFunction()868 auto* ctx = GetNodeContext(function_node); in UpdateFunction()961 NodeContext* ctx = GetNodeContext(node); in UpdateNode()964 ctx = CHECK_NOTNULL(GetNodeContext(node)); in UpdateNode()979 NodeContext* src_ctx = GetNodeContext(src); in UpdateNode()2104 auto ctx = refiner->GetNodeContext(&node); in ValidateSymbolicShapeManager()2163 auto ctx = refiner->GetNodeContext(&node); in VerboseShapeInferenceLogging()2380 auto* ctx = shape_refiner->GetNodeContext(queue_node); in UpdateQueue()2383 ctx = CHECK_NOTNULL(shape_refiner->GetNodeContext(queue_node)); in UpdateQueue()[all …]