Searched refs:local_gpu_pool (Results 1 – 1 of 1) sorted by relevance
2255 def local_gpu_pool(op, ctx_name, inputs, outputs): function2276 lifter = op_lifter([pool.Pool])(local_gpu_pool)2281 local_optimizer([pool.Pool])(local_gpu_pool),