Searched defs:TFE_CustomDeviceTensorHandleMethods (Results 1 – 1 of 1) sorted by relevance
553 typedef struct TFE_CustomDeviceTensorHandleMethods { struct554 int version = TFE_CUSTOM_DEVICE_VERSION;562 int (*num_dims)(void* data, TF_Status* status);565 int64_t (*dim)(void* data, int dim_index, TF_Status* status);567 void (*deallocator)(void* data);578 TF_Buffer* (*summarize)(void* data, TF_Status* status) = nullptr;