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