Searched defs:allow_mixing_unknown_and_cpu (Results 1 – 2 of 2) sorted by relevance
102 const jit::DeviceSet& devices, bool allow_mixing_unknown_and_cpu, in PickDeviceForXlaImpl()220 const jit::DeviceSet& devices, bool allow_mixing_unknown_and_cpu) { in PickDeviceForXla()230 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()