Home
last modified time | relevance | path

Searched defs:device_factory (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Ddevice_factory.h158 #define REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, ...) \ argument
162 #define INTERNAL_REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, \ argument
Ddevice_factory.cc249 auto device_factory = GetFactory(type); in NewDevice() local
/external/tensorflow/tensorflow/core/tpu/
Dtpu_global_init.cc60 DeviceFactory* device_factory = DeviceFactory::GetFactory(DEVICE_TPU_SYSTEM); in CreateDeviceMgr() local
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_test_util.cc114 auto device_factory = DeviceFactory::GetFactory("GPU"); in CreateGPUDevices() local