Searched refs:map_on_gpu (Results 1 – 3 of 3) sorted by relevance
260 def map_on_gpu(map_func): function
382 .apply(prefetching_ops.map_on_gpu(lambda _: cs.execute(fn))))
314 prefetching_ops.map_on_gpu(gpu_map_func))