コード例 #1
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]
コード例 #2
0
ファイル: convert_pkl_to_pb.py プロジェクト: Jhird/Detectron 
 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]