Searched refs:cpu_factory (Results 1 – 4 of 4) sorted by relevance
95 auto cpu_factory = GetFactory("CPU"); in ListAllPhysicalDevices() local96 if (!cpu_factory) { in ListAllPhysicalDevices()102 TF_RETURN_IF_ERROR(cpu_factory->ListPhysicalDevices(devices)); in ListAllPhysicalDevices()111 if (factory != cpu_factory) { in ListAllPhysicalDevices()128 auto cpu_factory = GetFactory("CPU"); in GetAnyDeviceDetails() local129 if (!cpu_factory) { in GetAnyDeviceDetails()135 TF_RETURN_IF_ERROR(cpu_factory->ListPhysicalDevices(&devices)); in GetAnyDeviceDetails()137 return cpu_factory->GetDeviceDetails(device_index, details); in GetAnyDeviceDetails()145 if (factory != cpu_factory) { in GetAnyDeviceDetails()165 auto cpu_factory = GetFactory("CPU"); in AddDevices() local[all …]
119 DeviceFactory* cpu_factory = DeviceFactory::GetFactory("CPU"); in GetStaticDeviceMgr() local121 auto status = cpu_factory->CreateDevices( in GetStaticDeviceMgr()
244 DeviceFactory* cpu_factory = DeviceFactory::GetFactory("CPU"); in RuntimeGraphOptimizer() local245 TF_RETURN_IF_ERROR(cpu_factory->CreateDevices( in RuntimeGraphOptimizer()
1184 DeviceFactory* cpu_factory = DeviceFactory::GetFactory("CPU"); in Run() local1185 if (!cpu_factory) { in Run()1189 TF_RETURN_IF_ERROR(cpu_factory->CreateDevices( in Run()