Searched refs:function_nodes (Results 1 – 2 of 2) sorted by relevance
165 std::unordered_set<const Node*> function_nodes; in InferShapesForFunction() local168 auto node_shape_inference_lambda = [this, &outer_context, &function_nodes, in InferShapesForFunction()173 function_nodes.insert(node); in InferShapesForFunction()183 for (const Node* node : function_nodes) { in InferShapesForFunction()
1069 std::vector<Node*> function_nodes; in MakePrunedGraphCopyAndInline() local1074 function_nodes.push_back(node); in MakePrunedGraphCopyAndInline()1077 for (auto node : function_nodes) { in MakePrunedGraphCopyAndInline()