Searched refs:graph_node (Results 1 – 4 of 4) sorted by relevance
56 auto graph_node = new (std::nothrow) GraphNode(graphDefT, opDef.get()); in Run() local57 if (graph_node == nullptr) { in Run()61 status = pass->Run(graph_node); in Run()62 delete graph_node; in Run()
1783 auto graph_node = MemoryProfiling::GetInstance().GetGraphMemoryNode(graph_id); in ConvertToProfilingNode() local1784 if (graph_node == nullptr) { in ConvertToProfilingNode()1785 graph_node = MemoryProfiling::GetInstance().AddGraphMemoryNode(graph_id); in ConvertToProfilingNode()1797 graph_node->AddTensorMemory(tensor_memory); in ConvertToProfilingNode()1814 graph_node->AddNodeMemory(node_memory); in ConvertToProfilingNode()
1374 auto graph_node = node->cast<CNodePtr>()->input(1)->cast<ValueNodePtr>(); in ProcessSubgraph() local1375 MS_EXCEPTION_IF_NULL(graph_node); in ProcessSubgraph()1376 FuncGraphPtr anf_graph = graph_node->value()->cast<FuncGraphPtr>(); in ProcessSubgraph()1382 std::string name = graph_node->ToString() + "_ge_graph.dot"; in ProcessSubgraph()
1711 auto graph_node = partial_cnode->input(kCallKernelGraphIndex); in GetCallSwitchKernelGraph() local1712 MS_EXCEPTION_IF_NULL(graph_node); in GetCallSwitchKernelGraph()1713 auto graph_value_node = graph_node->cast<ValueNodePtr>(); in GetCallSwitchKernelGraph()