Searched refs:function_nodes (Results 1 – 2 of 2) sorted by relevance
182 std::unordered_set<const Node*> function_nodes; in InferShapesForFunction() local185 auto node_shape_inference_lambda = [this, &outer_context, &function_nodes, in InferShapesForFunction()189 function_nodes.insert(node); in InferShapesForFunction()199 for (const Node* node : function_nodes) { in InferShapesForFunction()
1043 std::vector<Node*> function_nodes; in MakePrunedGraphCopyAndInline() local1048 function_nodes.push_back(node); in MakePrunedGraphCopyAndInline()1051 for (auto node : function_nodes) { in MakePrunedGraphCopyAndInline()