Searched refs:PickDeviceForXla (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_cluster_util_test.cc | 102 PickDeviceForXla(inputs_string, allow_mixing_unknown_and_cpu, &result)) in CheckPickDeviceResult()
116 PickDeviceForXla(inputs_string, allow_mixing_unknown_and_cpu, &result) in CheckPickDeviceHasError()
128 TEST(PickDeviceForXla, UniqueDevice) { in TEST() argument
132 TEST(PickDeviceForXla, DeviceOrder) { in TEST() argument
137 TEST(PickDeviceForXla, MultipleUnknownDevices) { in TEST() argument
141 TEST(PickDeviceForXla, GpuAndUnknown) { in TEST() argument
145 TEST(PickDeviceForXla, UnknownAndCpu) { in TEST() argument
149 TEST(PickDeviceForXla, MultipleDevicesOfSameType) { in TEST() argument
|
| D | xla_cluster_util.h | 118 Status PickDeviceForXla(absl::Span<const string> device_names,
|
| D | xla_cluster_util.cc | 337 Status PickDeviceForXla(absl::Span<const string> device_names, in PickDeviceForXla() function
|
| D | mark_for_compilation_pass.cc | 974 TF_RETURN_IF_ERROR(PickDeviceForXla( in AreDevicesCompatible()
1029 TF_RETURN_IF_ERROR(PickDeviceForXla( in ShouldCompileClusterImpl()
|
| D | build_xla_ops_pass.cc | 302 Status s = PickDeviceForXla(device_names_vector, true, result); in InferDeviceForCluster()
|