Searched refs:cpu_factory (Results 1 – 3 of 3) sorted by relevance
97 auto cpu_factory = GetFactory("CPU"); in AddDevices() local98 if (!cpu_factory) { in AddDevices()103 TF_RETURN_IF_ERROR(cpu_factory->CreateDevices(options, name_prefix, devices)); in AddDevices()112 if (factory != cpu_factory) { in AddDevices()
110 DeviceFactory* cpu_factory = DeviceFactory::GetFactory("CPU"); in OptimizeGraph() local111 TF_RETURN_IF_ERROR(cpu_factory->CreateDevices( in OptimizeGraph()
2548 DeviceFactory* cpu_factory = DeviceFactory::GetFactory("CPU"); in Run() local2549 if (!cpu_factory) { in Run()2553 TF_RETURN_IF_ERROR(cpu_factory->CreateDevices( in Run()