Searched refs:subgraph_node (Results 1 – 3 of 3) sorted by relevance
64 subgraph_node = graph.node_copy(70 subgraph_node.meta["source_fn_stack"] = [71 (subgraph_node, subgraph_node.target)73 decomposed_node_to_subgraph_node[decomposed_node] = subgraph_node
81 subgraph_node = graph.node_copy(87 subgraph_node.meta["source_fn_stack"] = [88 (subgraph_node, subgraph_node.target)91 subgraph_node
714 void ComputeSubgraphInMKLDNN(Node* subgraph_node) { in ComputeSubgraphInMKLDNN() argument715 auto graph = subgraph_node->owningGraph(); in ComputeSubgraphInMKLDNN()718 WithInsertPoint guard(subgraph_node); in ComputeSubgraphInMKLDNN()721 for (size_t i = 0; i < subgraph_node->inputs().size(); ++i) { in ComputeSubgraphInMKLDNN()722 Value* v = subgraph_node->inputs().at(i); in ComputeSubgraphInMKLDNN()728 ->insertBefore(subgraph_node); in ComputeSubgraphInMKLDNN()731 subgraph_node->replaceInput(i, to_mkldnn->output()); in ComputeSubgraphInMKLDNN()734 for (size_t i = 0; i < subgraph_node->outputs().size(); ++i) { in ComputeSubgraphInMKLDNN()735 Value* v = subgraph_node->outputs().at(i); in ComputeSubgraphInMKLDNN()743 ->insertAfter(subgraph_node); in ComputeSubgraphInMKLDNN()[all …]