Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dcustom_device_testutil.cc131 TFE_TensorHandleDevicePointer(input, s)); in LoggingDeviceExecute()
182 TFE_TensorHandleDevicePointer(logged_tensor_handle, status)) in UnpackTensorHandle()
H A Ddlpack.cc276 auto* tf_dlm_data = TFE_TensorHandleDevicePointer(h, status); in TFE_HandleToDLPack()
H A Dc_api_experimental.h408 TF_CAPI_EXPORT extern void* TFE_TensorHandleDevicePointer(TFE_TensorHandle*,
H A Dc_api_experimental_test.cc390 void* data = TFE_TensorHandleDevicePointer(copy, status); in TEST()
H A Dc_api.cc369 void* TFE_TensorHandleDevicePointer(TFE_TensorHandle* h, TF_Status* status) { in TFE_TensorHandleDevicePointer() function
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/parallel_device/
H A Dparallel_device.cc294 TFE_TensorHandleDevicePointer(tensor, status)); in CopyTensorFromParallelDevice()
353 TFE_TensorHandleDevicePointer(input, status))); in ParallelDeviceExecute()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device.cc596 TFE_TensorHandleDevicePointer(input, status)); in FetchLayout()
618 TFE_TensorHandleDevicePointer(input, status)); in Unpack()
713 TFE_TensorHandleDevicePointer(input_tensor, status)); in CopyToMesh()
1072 TFE_TensorHandleDevicePointer(input, status)); in IsSparseDTensor()
1884 TFE_TensorHandleDevicePointer(input, status)); in Execute()
1987 TFE_TensorHandleDevicePointer(tensor, status)); in CopyFromDTensorDevice()
/aosp_15_r20/external/tensorflow/tensorflow/security/advisory/
H A Dtfsa-2020-025.md14 dlm_tensor->dl_tensor.data = TFE_TensorHandleDevicePointer(h, status);
H A Dtfsa-2020-024.md24 dlm_tensor->dl_tensor.data = TFE_TensorHandleDevicePointer(h, status);