Searched refs:function_nodes (Results 1 – 2 of 2) sorted by relevance
157 std::unordered_set<const Node*> function_nodes; in InferShapesForFunction() local160 auto node_shape_inference_lambda = [this, &outer_context, &function_nodes, in InferShapesForFunction()165 function_nodes.insert(node); in InferShapesForFunction()188 for (const Node* node : function_nodes) { in InferShapesForFunction()203 for (const Node* node : function_nodes) { in InferShapesForFunction()
2307 std::vector<Node*> function_nodes; in MakePrunedGraphCopyAndInline() local2312 function_nodes.push_back(node); in MakePrunedGraphCopyAndInline()2315 for (auto node : function_nodes) { in MakePrunedGraphCopyAndInline()