Searched refs: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::CustomDeviceAPI420 ~CustomDeviceAPI() override { device_.delete_device(info_); } in ~CustomDeviceAPI()1145 auto custom_device = std::make_unique<tensorflow::CustomDeviceAPI>( in TFE_RegisterCustomDevice()