Searched refs:DLManagedTensor (Results 1 – 9 of 9) sorted by relevance
14 TORCH_API DLManagedTensor* toDLPack(const Tensor& src);15 TORCH_API Tensor fromDLPack(DLManagedTensor* src);17 inline Tensor fromDLPack(const DLManagedTensor* src) { in fromDLPack()18 return fromDLPack(const_cast<DLManagedTensor*>(src)); in fromDLPack()21 fromDLPack(DLManagedTensor* src, std::function<void(void*)> deleter);
217 typedef struct DLManagedTensor { struct229 void (*deleter)(struct DLManagedTensor * self); argument230 } DLManagedTensor; typedef
258 DLManagedTensor tensor;261 static void deleter(DLManagedTensor* arg) { in deleter()267 DLManagedTensor* toDLPack(const Tensor& src) { in toDLPack()297 Tensor fromDLPack(DLManagedTensor* src) { in fromDLPack()306 Tensor fromDLPack(DLManagedTensor* src, std::function<void(void*)> deleter) { in fromDLPack()
16 DLManagedTensor* dlMTensor = toDLPack(a); in TEST()27 DLManagedTensor* dlMTensor = toDLPack(a); in TEST()41 DLManagedTensor* dlMTensor = toDLPack(a); in TEST()
16 DLManagedTensor* dlMTensor = toDLPack(a); in TEST()27 DLManagedTensor* dlMTensor = toDLPack(a); in TEST()
41 DLManagedTensor tensor;68 void DLManagedTensorDeleter(DLManagedTensor* arg) { in DLManagedTensorDeleter()264 DLManagedTensor* dlMTensor = static_cast<DLManagedTensor*>(dlm_ptr); in TFE_CallDLManagedTensorDeleter()293 DLManagedTensor* dlm_tensor = &tf_dlm_tensor_ctx->tensor; in TFE_HandleToDLPack()328 DLManagedTensor* dlmt = static_cast<DLManagedTensor*>(dlm); in TFE_HandleFromDLPack()
53 DLManagedTensor tensor;63 void DLPackTensorDeleter(DLManagedTensor* t) { in DLPackTensorDeleter()325 DLManagedTensor* dlmt = static_cast<DLManagedTensor*>( in BufferToDLPackManagedTensor()358 DLManagedTensor* dlmt = static_cast<DLManagedTensor*>(tensor); in DLPackManagedTensorToBuffer()
583 DLManagedTensor* dlMTensor = in DLPack_Capsule_Destructor()584 (DLManagedTensor*)PyCapsule_GetPointer(data, "dltensor"); in DLPack_Capsule_Destructor()595 DLManagedTensor* dlMTensor = at::toDLPack(THPVariable_Unpack(data)); in THPModule_toDLPack()
1639 DLManagedTensor* dlMTensor = in tensor_fromDLPack()1640 (DLManagedTensor*)PyCapsule_GetPointer(data, "dltensor"); in tensor_fromDLPack()