Searched defs:custom_device (Results 1 – 6 of 6) sorted by relevance
165 TFE_CustomDevice custom_device; in RegisterLoggingDevice() local189 TFE_CustomDevice* custom_device = new TFE_CustomDevice; in AllocateLoggingDevice() local
1145 auto custom_device = std::make_unique<tensorflow::CustomDeviceAPI>( in TFE_RegisterCustomDevice() local
60 if (auto* custom_device = fallback_request_state.custom_device(device)) { in GetDeviceFromFallbackState() local
98 tensorflow::Device* custom_device(const tensorflow::Device* device) const { in custom_device() function
57 tensorflow::CustomDevice* custom_device = nullptr; in Execute() local
153 TestCustomDevice custom_device(custom_device_name); in TEST() local