Searched refs:TFE_TensorHandleDevicePointer (Results 1 – 8 of 8) sorted by relevance
123 TFE_TensorHandleDevicePointer(input, s)); in LoggingDeviceExecute()174 TFE_TensorHandleDevicePointer(logged_tensor_handle, status)) in UnpackTensorHandle()
257 auto* tf_dlm_data = TFE_TensorHandleDevicePointer(h, status); in TFE_HandleToDLPack()
378 TF_CAPI_EXPORT extern void* TFE_TensorHandleDevicePointer(TFE_TensorHandle*,
387 void* data = TFE_TensorHandleDevicePointer(copy, status); in TEST()
364 void* TFE_TensorHandleDevicePointer(TFE_TensorHandle* h, TF_Status* status) { in TFE_TensorHandleDevicePointer() function
14 dlm_tensor->dl_tensor.data = TFE_TensorHandleDevicePointer(h, status);
24 dlm_tensor->dl_tensor.data = TFE_TensorHandleDevicePointer(h, status);
310 TFE_TensorHandleDevicePointer(input, status))); in ParallelDeviceExecute()