Searched refs:fun_node (Results 1 – 1 of 1) sorted by relevance
791 NodeDef* fun_node = gv.GetNode(fun_input.node_name); in UpdateFunction() local821 (*fun_node->mutable_attr())["_output_shapes"] = output_attr; in UpdateFunction()838 (*fun_node->mutable_attr())["_handle_dtypes"] = dtype_attr; in UpdateFunction()839 (*fun_node->mutable_attr())["_handle_shapes"] = shape_attr; in UpdateFunction()