| 127 | } |
| 128 | |
| 129 | void CPUDriver::SetGraphDoneHook(Device* dev, void* graph_handle, dev_graph_cb_t func) |
| 130 | { |
| 131 | DevContext* context = reinterpret_cast<DevContext*>(graph_handle); |
| 132 | context->graph_cb = func; |
| 133 | } |
| 134 | |
| 135 | Subgraph* CPUDriver::GetOptimizedGraph(Device* dev, void* graph_handle) |
| 136 | { |