Searched refs:dlcapsule (Results 1 – 3 of 3) sorted by relevance
69 dlcapsule = dlpack.to_dlpack(tf_tensor)71 tf_tensor2 = dlpack.from_dlpack(dlcapsule)84 dlcapsule = dlpack.to_dlpack(tf_tensor)86 _ = dlpack.from_dlpack(dlcapsule)89 dlpack.from_dlpack(dlcapsule) # Should can be consumed only once
49 def from_dlpack(dlcapsule): argument66 return pywrap_tfe.TFE_FromDlpackCapsule(dlcapsule, context.context()._handle)
5 argspec: "args=[\'dlcapsule\'], varargs=None, keywords=None, defaults=None"