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