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)