Searched refs:flatten_reshape_node (Results 1 – 1 of 1) sorted by relevance
1052 Node* flatten_reshape_node = nullptr; in InsertReshapeNodePairs() local1060 .Finalize(graph, &flatten_reshape_node)); in InsertReshapeNodePairs()1082 Node* recover_reshape_input_node = flatten_reshape_node; in InsertReshapeNodePairs()1115 graph->AddEdge(flatten_reshape_node, 0, edge->dst(), edge->dst_input()); in InsertReshapeNodePairs()1137 for (const Edge* out_edge : flatten_reshape_node->out_edges()) { in InsertReshapeNodePairs()