Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Ddevice_factory.h136 #define REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, ...) \ argument
140 #define INTERNAL_REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, \ argument
Ddevice_factory.cc191 auto device_factory = GetFactory(type); in NewDevice() local
/external/tensorflow/tensorflow/core/common_runtime/
Dpermuter_test.cc122 auto device_factory = DeviceFactory::GetFactory("GPU"); in InitGPUDevices() local
Dring_gatherer_test.cc123 auto device_factory = DeviceFactory::GetFactory("GPU"); in InitGPUDevices() local
Dring_reducer_test.cc146 auto device_factory = DeviceFactory::GetFactory("GPU"); in InitGPUDevices() local
Dhierarchical_tree_broadcaster_test.cc212 auto device_factory = DeviceFactory::GetFactory("GPU"); in InitGPUDevices() local