because the tensors are loaded asynchronously.

  }];

  let arguments = (ins
    TFTensorType:$prefix,
    TFTensorType:$tensor_names,
    TFTensorType:$shape_and_slices,
    Variadic<TFTensorType>:$var_handles,
    TypeArrayAttr: $restored_dtypes,
    DenseBoolArrayAttr: $truncate_in_cast
  );
}

  } else {
    return first_try_buffer.status();
  }
}
}  // namespace

void PjRtDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor,
                                              absl::string_view tensor_name,
                                              Device* device,
                                              Tensor* cpu_tensor,
                                              StatusCallback done) {

    });
    if (!status.ok()) {
      done(status);
    }
  }
}

void XlaDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor,
                                             absl::string_view tensor_name,
                                             Device* device, Tensor* cpu_tensor,
                                             StatusCallback done) {
  if (device_tensor->NumElements() == 0) {