Searched defs:keep_graph (Results 1 – 7 of 7) sorted by relevance
126 bool keep_graph, in execute()191 unsigned char keep_graph = 0; in THPEngine_run_backward() local
93 bool keep_graph, in run_backward()
659 bool keep_graph, in GraphTask()1191 bool keep_graph, in execute()
585 std::optional<bool> keep_graph, in _backward()
27 …ensorList inputs, const std::optional<Tensor>& gradient_opt, std::optional<bool> keep_graph, bool … in _backward()
138 std::optional<bool> keep_graph, in _backward()
1138 bool keep_graph = retain_graph ? retain_graph.value() : create_graph; in __anonbfe5918f4602() local