Searched defs:TFE_CustomDeviceTensorHandleMethods (Results 1 – 1 of 1) sorted by relevance
554 typedef struct TFE_CustomDeviceTensorHandleMethods { struct555 int version = TFE_CUSTOM_DEVICE_VERSION;563 int (*num_dims)(void* data, TF_Status* status);566 int64_t (*dim)(void* data, int dim_index, TF_Status* status);568 void (*deallocator)(void* data);579 TF_Buffer* (*summarize)(void* data, TF_Status* status) = nullptr;