Searched refs:device_allows_sync_on_completion (Results 1 – 1 of 1) sorted by relevance
253 const bool device_allows_sync_on_completion = in CopyDeviceTensorToCPU() local260 device_allows_sync_on_completion](xla::Status status) { in CopyDeviceTensorToCPU()268 if (!device_allows_sync_on_completion) { in CopyDeviceTensorToCPU()277 if (!device_allows_sync_on_completion && in CopyDeviceTensorToCPU()