Searched refs:PickDeviceForXla (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | device_util_test.cc | 36 PickDeviceForXla(cache, device_set, allow_mixing_unknown_and_cpu)); in PickDeviceHelper()
72 TEST(PickDeviceForXla, UniqueDevice) { in TEST() argument
76 TEST(PickDeviceForXla, MoreSpecificDevice) { in TEST() argument
83 TEST(PickDeviceForXla, DeviceOrder) { in TEST() argument
89 TEST(PickDeviceForXla, MultipleUnknownDevices) { in TEST() argument
93 TEST(PickDeviceForXla, GpuAndUnknown) { in TEST() argument
97 TEST(PickDeviceForXla, UnknownAndCpu) { in TEST() argument
101 TEST(PickDeviceForXla, MultipleDevicesOfSameType) { in TEST() argument
|
| D | device_util.h | 201 xla::StatusOr<jit::DeviceId> PickDeviceForXla(
|
| D | device_util.cc | 218 xla::StatusOr<jit::DeviceId> PickDeviceForXla( in PickDeviceForXla() function
|
| D | build_xla_ops_pass.cc | 373 PickDeviceForXla(*device_info_cache, device_set, in InferDeviceForCluster()
|
| D | mark_for_compilation_pass.cc | 1628 PickDeviceForXla(device_info_cache_, cluster.devices(), in ShouldCompileClusterImpl()
|