Searched refs:allow_mixing_unknown_and_cpu (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_cluster_util_test.cc | 95 bool allow_mixing_unknown_and_cpu, in CheckPickDeviceResult() argument
102 PickDeviceForXla(inputs_string, allow_mixing_unknown_and_cpu, &result)) in CheckPickDeviceResult()
104 << "], allow_mixing_unknown_and_cpu=" << allow_mixing_unknown_and_cpu in CheckPickDeviceResult()
109 void CheckPickDeviceHasError(bool allow_mixing_unknown_and_cpu, in CheckPickDeviceHasError() argument
116 PickDeviceForXla(inputs_string, allow_mixing_unknown_and_cpu, &result) in CheckPickDeviceHasError()
|
| D | xla_cluster_util.cc | 252 bool allow_mixing_unknown_and_cpu, in PickDeviceForXlaImpl() argument
314 if (!allow_mixing_unknown_and_cpu) { in PickDeviceForXlaImpl()
338 bool allow_mixing_unknown_and_cpu, in PickDeviceForXla() argument
340 return PickDeviceForXlaImpl(device_names, allow_mixing_unknown_and_cpu, in PickDeviceForXla()
346 bool allow_mixing_unknown_and_cpu, in CanPickDeviceForXla() argument
348 return PickDeviceForXlaImpl(device_names, allow_mixing_unknown_and_cpu, in CanPickDeviceForXla()
|
| D | xla_cluster_util.h | 119 bool allow_mixing_unknown_and_cpu,
126 bool allow_mixing_unknown_and_cpu,
|