Searched defs:allow_mixing_unknown_and_cpu (Results 1 – 2 of 2) sorted by relevance
101 const jit::DeviceSet& devices, bool allow_mixing_unknown_and_cpu, in PickDeviceForXlaImpl()219 const jit::DeviceSet& devices, bool allow_mixing_unknown_and_cpu) { in PickDeviceForXla()229 const jit::DeviceSet& devices, bool allow_mixing_unknown_and_cpu) { in MaybePickDeviceForXla()
24 Status PickDeviceHelper(bool allow_mixing_unknown_and_cpu, in PickDeviceHelper()42 bool allow_mixing_unknown_and_cpu, in CheckPickDeviceResult()52 void CheckPickDeviceHasError(bool allow_mixing_unknown_and_cpu, in CheckPickDeviceHasError()