Ejemplo n.º 1
0
 def is_on_host(f):
     return not is_on_device(f, options['gpu-fit'])
Ejemplo n.º 2
0
 def callback(f):
     if not is_on_device(f, options['gpu-fit']):
         return [f.time_dim]
     else:
         return None