Searched defs:TFE_CustomDevice (Results 1 – 1 of 1) sorted by relevance
441 typedef struct TFE_CustomDevice { struct442 int version = TFE_CUSTOM_DEVICE_VERSION;444 TFE_TensorHandle* (*copy_tensor_to_device)(TFE_Context* context,450 TFE_TensorHandle* (*copy_tensor_from_device)(TFE_Context* context,481 } TFE_CustomDevice; argument