Home
last modified time | relevance | path

Searched refs:device_factory (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Ddevice_factory.h136 #define REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, ...) \ argument
137 INTERNAL_REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, \
140 #define INTERNAL_REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, \ argument
142 static ::tensorflow::dfactory::Registrar<device_factory> \
Ddevice_factory.cc191 auto device_factory = GetFactory(type); in NewDevice() local
192 if (!device_factory) { in NewDevice()
198 TF_CHECK_OK(device_factory->CreateDevices(opt, name_prefix, &devices)); in NewDevice()
DBUILD57 "device_factory.h",
189 "device_factory.h",
264 "device_factory.cc",
358 "device_factory.cc",
359 "device_factory.h",
/external/tensorflow/tensorflow/core/kernels/mlir_generated/
DBUILD385 "//tensorflow/core/common_runtime:device_factory",
397 "//tensorflow/core/common_runtime:device_factory",
414 "//tensorflow/core/common_runtime:device_factory",
436 "//tensorflow/core/common_runtime:device_factory",
448 "//tensorflow/core/common_runtime:device_factory",
/external/tensorflow/tensorflow/core/common_runtime/
Dpermuter_test.cc122 auto device_factory = DeviceFactory::GetFactory("GPU"); in InitGPUDevices() local
123 CHECK(device_factory); in InitGPUDevices()
125 Status s = device_factory->CreateDevices( in InitGPUDevices()
DBUILD59 "device_factory.h",
155 "device_factory.h",
457 ":device_factory",
506 name = "device_factory",
507 hdrs = ["device_factory.h"],
607 ":device_factory",
763 ":device_factory",
1535 ":device_factory",
1553 ":device_factory",
1582 ":device_factory",
Dring_gatherer_test.cc123 auto device_factory = DeviceFactory::GetFactory("GPU"); in InitGPUDevices() local
124 CHECK(device_factory); in InitGPUDevices()
126 Status s = device_factory->CreateDevices( in InitGPUDevices()
Dring_reducer_test.cc146 auto device_factory = DeviceFactory::GetFactory("GPU"); in InitGPUDevices() local
147 CHECK(device_factory); in InitGPUDevices()
149 Status s = device_factory->CreateDevices( in InitGPUDevices()
Dhierarchical_tree_broadcaster_test.cc212 auto device_factory = DeviceFactory::GetFactory("GPU"); in InitGPUDevices() local
213 CHECK(device_factory); in InitGPUDevices()
215 Status s = device_factory->CreateDevices( in InitGPUDevices()
/external/tensorflow/tensorflow/tools/def_file_filter/
Dsymbols_pybind.txt77 [//tensorflow/core/common_runtime:device_factory] # device_lib, tfe, tf_session
/external/tensorflow/tensorflow/compiler/jit/
DBUILD188 "//tensorflow/core/common_runtime:device_factory",
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/
DBUILD1093 "//tensorflow/core/common_runtime:device_factory",
/external/tensorflow/tensorflow/core/
DBUILD423 "//tensorflow/core/framework:device_factory.h",