Searched refs:dl_tensor (Results 1 – 4 of 4) sorted by relevance
277 dlm_tensor->dl_tensor.ctx = tf_dlm_context; in TFE_HandleToDLPack()279 dlm_tensor->dl_tensor.ndim = ndim; in TFE_HandleToDLPack()280 dlm_tensor->dl_tensor.data = tf_dlm_data; in TFE_HandleToDLPack()281 dlm_tensor->dl_tensor.dtype = tf_dlm_type; in TFE_HandleToDLPack()294 dlm_tensor->dl_tensor.shape = shape_arr->data(); in TFE_HandleToDLPack()300 dlm_tensor->dl_tensor.strides = stride_arr->data(); in TFE_HandleToDLPack()302 dlm_tensor->dl_tensor.byte_offset = in TFE_HandleToDLPack()310 DLTensor* dl_tensor = &dlmt->dl_tensor; in TFE_HandleFromDLPack() local312 DeviceNameFromDlContext(dl_tensor->ctx, status); in TFE_HandleFromDLPack()319 Status s = TfDataTypeFormDlDataType(dl_tensor->dtype, &dtype); in TFE_HandleFromDLPack()[all …]
260 DLTensor& dt = pack->tensor.dl_tensor; in BufferToDLPackManagedTensor()327 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()344 reinterpret_cast<int64*>(dlmt->dl_tensor.strides), in DLPackManagedTensorToBuffer()345 dlmt->dl_tensor.ndim); in DLPackManagedTensorToBuffer()348 minor_to_major.resize(dlmt->dl_tensor.ndim); in DLPackManagedTensorToBuffer()[all …]
14 dlm_tensor->dl_tensor.data = TFE_TensorHandleDevicePointer(h, status);15 dlm_tensor->dl_tensor.dtype = GetDlDataType(data_type, status);21 dlm_tensor->dl_tensor.shape = &(*shape_arr)[0];22 dlm_tensor->dl_tensor.strides = &(*stride_arr)[0];
24 dlm_tensor->dl_tensor.data = TFE_TensorHandleDevicePointer(h, status);25 dlm_tensor->dl_tensor.dtype = GetDlDataType(data_type, status);