Searched refs:device_allows_sync_on_completion (Results 1 – 1 of 1) sorted by relevance
255 const bool device_allows_sync_on_completion = in CopyDeviceTensorToCPU() local262 device_allows_sync_on_completion](xla::Status status) { in CopyDeviceTensorToCPU()270 if (!device_allows_sync_on_completion) { in CopyDeviceTensorToCPU()279 if (!device_allows_sync_on_completion && in CopyDeviceTensorToCPU()