Searched refs:root_on_gpu (Results 1 – 1 of 1) sorted by relevance
1199 bool root_on_gpu = NodeIsOnGpu(node_def); in FindMatMulBiasAddAndGelu() local1201 root_on_gpu ? &subgraph_gpu : &subgraph_cpu; in FindMatMulBiasAddAndGelu()