Searched defs:dlMTensor (Results 1 – 5 of 5) sorted by relevance
16 DLManagedTensor* dlMTensor = toDLPack(a); in TEST() local27 DLManagedTensor* dlMTensor = toDLPack(a); in TEST() local41 DLManagedTensor* dlMTensor = toDLPack(a); in TEST() local
16 DLManagedTensor* dlMTensor = toDLPack(a); in TEST() local27 DLManagedTensor* dlMTensor = toDLPack(a); in TEST() local
264 DLManagedTensor* dlMTensor = static_cast<DLManagedTensor*>(dlm_ptr); in TFE_CallDLManagedTensorDeleter() local
583 DLManagedTensor* dlMTensor = in DLPack_Capsule_Destructor() local595 DLManagedTensor* dlMTensor = at::toDLPack(THPVariable_Unpack(data)); in THPModule_toDLPack() local
1639 DLManagedTensor* dlMTensor = in tensor_fromDLPack() local