Searched refs:CustomDeviceAPI (Results 1 – 1 of 1) sorted by relevance
409 class CustomDeviceAPI : public tensorflow::CustomDevice { class411 CustomDeviceAPI(TFE_Context* context, TFE_CustomDevice device, void* info, in CustomDeviceAPI() function in tensorflow::__anon6f1bef310211::CustomDeviceAPI415 ~CustomDeviceAPI() override { device_.delete_device(info_); } in ~CustomDeviceAPI()1153 auto custom_device = std::make_unique<tensorflow::CustomDeviceAPI>( in TFE_RegisterCustomDevice()