Searched refs:keep_graph_ (Results 1 – 4 of 4) sorted by relevance
49 sharedContext->retrieveGraphTask()->keep_graph_); in apply()
25 bool keep_graph_; member
418 return current_graph_task ? current_graph_task->keep_graph_ : true; in get_current_graph_task_keep_graph()665 : keep_graph_(keep_graph), in GraphTask()950 if (!graph_task->keep_graph_) { in call_function()1042 if (!graph_task->keep_graph_) { in evaluate_function()
674 if (!graph_task.keep_graph_) { in _compiled_autograd_impl()