Searched refs:CreateAutodiffSubgraphs (Results 1 – 6 of 6) sorted by relevance
14 TORCH_API std::vector<Node*> CreateAutodiffSubgraphs(
462 std::vector<Node*> CreateAutodiffSubgraphs( in CreateAutodiffSubgraphs() function
12 CreateAutodiffSubgraphs(graph, /*threshold=*/2); in TEST()
493 auto diff_nodes = CreateAutodiffSubgraphs( in runProfilingOptimizations()
743 auto diff_nodes = CreateAutodiffSubgraphs( in compileSpec()
676 CreateAutodiffSubgraphs(graph); in initJITBindings()678 CreateAutodiffSubgraphs(graph, py::cast<int>(threshold)); in initJITBindings()