Searched refs:grad_with_create_graph (Results 1 – 1 of 1) sorted by relevance
991 def grad_with_create_graph(mod, x, target): function1005 compiled_f1 = torch.compile(backend="aot_eager")(grad_with_create_graph)