Home
last modified time | relevance | path

Searched refs:wrapper_graph (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/torch/csrc/jit/codegen/fuser/
Dinterface.cpp58 auto wrapper_graph = std::make_shared<Graph>(); in debugLaunchGraph() local
59 Node* fusion_group = wrapper_graph->insertNode( in debugLaunchGraph()
60 wrapper_graph->createWithSubgraph(prim::FusionGroup)); in debugLaunchGraph()
63 fusion_group->addInput(wrapper_graph->addInput()); in debugLaunchGraph()
66 wrapper_graph->registerOutput(fusion_group->addOutput()); in debugLaunchGraph()
80 auto wrapper_graph = std::make_shared<Graph>(); in debugGetFusedKernelCode() local
81 Node* fusion_group = wrapper_graph->insertNode( in debugGetFusedKernelCode()
82 wrapper_graph->createWithSubgraph(prim::FusionGroup)); in debugGetFusedKernelCode()
85 fusion_group->addInput(wrapper_graph->addInput()); in debugGetFusedKernelCode()
88 wrapper_graph->registerOutput(fusion_group->addOutput()); in debugGetFusedKernelCode()