Home
last modified time | relevance | path

Searched refs:TFE_CustomDeviceTensorHandleMethods (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api.cc497 TFE_CustomDeviceTensorHandleMethods methods) in CAPICustomDeviceTensorHandle()
536 const TFE_CustomDeviceTensorHandleMethods methods_;
544 TFE_CustomDeviceTensorHandleMethods methods, TF_Status* status) { in TFE_NewCustomDeviceTensorHandle()
H A Dcustom_device_testutil.cc66 TFE_CustomDeviceTensorHandleMethods handle_methods; in MakeLoggedTensorHandle()
H A Dc_api_experimental.h554 typedef struct TFE_CustomDeviceTensorHandleMethods { struct
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/parallel_device/
H A Dparallel_device.cc250 TFE_CustomDeviceTensorHandleMethods handle_methods; in ParallelTensorToTensorHandle()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device.cc513 TFE_CustomDeviceTensorHandleMethods handle_methods; in MakeLayoutTensorHandle()