- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for CanShapedBufferBeAccessedNow (0.57 sec)
-
tensorflow/compiler/jit/xla_device_context.cc
VLOG(2) << "Transfer to device as literal: " << literal.ToString() << " " << xla_tensor->shaped_buffer().ToString(); if (UseMultipleStreams() && !transfer_manager_->CanShapedBufferBeAccessedNow( stream_->parent(), xla_tensor->shaped_buffer())) { // Initially wait for the compute stream so that memory allocations are // synchronized.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 12.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_tpu_device.cc
// Wait for the destination tensor buffers to be ready, if they are not // available for an immediate write. if (!dst_xla_context->transfer_manager()->CanShapedBufferBeAccessedNow( dst_compute_stream->parent(), xla_output->shaped_buffer())) { TF_RETURN_IF_ERROR( dst_device_to_device_stream->WaitFor(dst_compute_stream));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 20.9K bytes - Viewed (0)