def zerocopy_from_dgl_ndarray(arr):
return nd.from_dlpack(arr.to_dlpack())
def mxnet_adapter(fun, in1, in2):
tin1 = [mxnd.from_dlpack(dltensor) for dltensor in in1]
tin2 = [mxnd.from_dlpack(dltensor) for dltensor in in2]
tout1, tout2 = fun(tin1, tin2)
return [mxnd.to_dlpack_for_read(tout) for tout in tout1], \
[mxnd.to_dlpack_for_read(tout) for tout in tout2]
def zerocopy_from_dlpack(dlpack_arr):
return nd.from_dlpack(dlpack_arr)
