def convert_op_gpu(op):
     for x in CPU_OPS:
         if mutils.filter_op(op, type=x[0], inputs=x[1]):
             return None
     op.device_option.CopyFrom(cdo_cuda)
     return [op]
Ejemplo n.º 2
0
 def convert_op_gpu(op):
     for x in CPU_OPS:
         if mutils.filter_op(op, type=x[0], inputs=x[1]):
             return None
     op.device_option.CopyFrom(cdo_cuda)
     return [op]