Searched refs:GraphRoot (Results 1 – 5 of 5) sorted by relevance
69 void GraphRoot::compiled_args(CompiledNodeArgs& args) { in compiled_args()72 variable_list GraphRoot::apply_with_saved( in apply_with_saved()
80 struct TORCH_API GraphRoot : public Node { struct81 GraphRoot(edge_list functions, variable_list inputs) in GraphRoot() function
49 const ExplodedNode *GraphRoot = *G.roots_begin(); in checkEndAnalysis() local50 const LocationContext *LC = GraphRoot->getLocation().getLocationContext(); in checkEndAnalysis()
20 using torch::autograd::GraphRoot;307 GraphRoot dummyRoot(edges, {}); in computeDependencies()483 auto dummyRoot = std::make_shared<GraphRoot>(edge_list(), variable_list()); in executeSendFunctionAsync()575 std::make_shared<GraphRoot>(rootEdges, grads); in execute()
1245 : std::make_shared<GraphRoot>(root_edges, inputs); in execute()