Searched defs:CustomDeviceAPI (Results 1 – 1 of 1) sorted by relevance
414 class CustomDeviceAPI : public tensorflow::CustomDevice { class416 CustomDeviceAPI(TFE_Context* context, TFE_CustomDevice device, void* info, in CustomDeviceAPI() function in tensorflow::__anon816881a00211::CustomDeviceAPI