Searched defs:TFE_CustomDevice (Results 1 – 1 of 1) sorted by relevance
473 typedef struct TFE_CustomDevice { struct474 int version = TFE_CUSTOM_DEVICE_VERSION;476 TFE_TensorHandle* (*copy_tensor_to_device)(TFE_Context* context,482 TFE_TensorHandle* (*copy_tensor_from_device)(TFE_Context* context,513 } TFE_CustomDevice; argument