Searched refs:system_device (Results 1 – 3 of 3) sorted by relevance
31 DeviceNameUtils::ParsedName* system_spec, Device** system_device) { in GetSystemDevice() argument85 *system_device = system_devices[0]; in GetSystemDevice()86 if (!DeviceNameUtils::ParseFullName((*system_device)->name(), system_spec)) { in GetSystemDevice()88 (*system_device)->name(), in GetSystemDevice()
42 Device** system_device);
166 std::string GetTPUCompilationDevice(Device system_device) { in GetTPUCompilationDevice() argument171 system_device.type = tensorflow::DEVICE_CPU; in GetTPUCompilationDevice()172 return DeviceNameUtils::ParsedNameToString(system_device); in GetTPUCompilationDevice()