예제 #1
0
 def zerocopy_from_dlpack(input):
     return tfdlpack.from_dlpack(input)
예제 #2
0
파일: tensor.py 프로젝트: hacors/Drug 
def zerocopy_from_dlpack(dlpack_tensor):
    return tfdlpack.from_dlpack(dlpack_tensor)