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