Home
last modified time | relevance | path

Searched refs:TFE_TensorHandleDevicePointer (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dcustom_device_testutil.cc131 TFE_TensorHandleDevicePointer(input, s)); in LoggingDeviceExecute()
182 TFE_TensorHandleDevicePointer(logged_tensor_handle, status)) in UnpackTensorHandle()
Ddlpack.cc257 auto* tf_dlm_data = TFE_TensorHandleDevicePointer(h, status); in TFE_HandleToDLPack()
Dc_api_experimental.h407 TF_CAPI_EXPORT extern void* TFE_TensorHandleDevicePointer(TFE_TensorHandle*,
Dc_api_experimental_test.cc386 void* data = TFE_TensorHandleDevicePointer(copy, status); in TEST()
Dc_api.cc367 void* TFE_TensorHandleDevicePointer(TFE_TensorHandle* h, TF_Status* status) { in TFE_TensorHandleDevicePointer() function
/external/tensorflow/tensorflow/security/advisory/
Dtfsa-2020-025.md14 dlm_tensor->dl_tensor.data = TFE_TensorHandleDevicePointer(h, status);
Dtfsa-2020-024.md24 dlm_tensor->dl_tensor.data = TFE_TensorHandleDevicePointer(h, status);
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device.cc294 TFE_TensorHandleDevicePointer(tensor, status)); in CopyTensorFromParallelDevice()
353 TFE_TensorHandleDevicePointer(input, status))); in ParallelDeviceExecute()