Searched refs:dlmt (Results 1 – 2 of 2) sorted by relevance
305 DLManagedTensor* dlmt = static_cast<DLManagedTensor*>( in BufferToDLPackManagedTensor() local307 if (dlmt) { in BufferToDLPackManagedTensor()308 DLPackTensorDeleter(dlmt); in BufferToDLPackManagedTensor()326 DLManagedTensor* dlmt = static_cast<DLManagedTensor*>(tensor); in DLPackManagedTensorToBuffer() local327 if (dlmt->dl_tensor.ndim < 0) { in DLPackManagedTensorToBuffer()330 dlmt->dl_tensor.ndim); in DLPackManagedTensorToBuffer()334 DeviceForDLContext(*client->pjrt_client(), dlmt->dl_tensor.ctx)); in DLPackManagedTensorToBuffer()336 reinterpret_cast<int64*>(dlmt->dl_tensor.shape), dlmt->dl_tensor.ndim); in DLPackManagedTensorToBuffer()338 DLDataTypeToPrimitiveType(dlmt->dl_tensor.dtype)); in DLPackManagedTensorToBuffer()341 if (dlmt->dl_tensor.strides && in DLPackManagedTensorToBuffer()[all …]
309 DLManagedTensor* dlmt = static_cast<DLManagedTensor*>(dlm); in TFE_HandleFromDLPack() local310 DLTensor* dl_tensor = &dlmt->dl_tensor; in TFE_HandleFromDLPack()343 total_bytes, &DeallocatorWrapperFunc, dlmt, status); in TFE_HandleFromDLPack()