- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for DoHostCallback (0.2 sec)
-
tensorflow/compiler/jit/xla_device_context.cc
// out the same buffers until the transfer has completed. status = host_to_device_stream_->DoHostCallback([ref]() { ref.Unref(); }); done(status); } else { status = host_to_device_stream_->DoHostCallback([ref, done]() { ref.Unref(); done(absl::OkStatus()); }); if (!status.ok()) { done(status); } } }
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
// might not be enqueued to the stream yet, we put it on destination stream. TensorReference input_reference(*input); std::move(return_substream).release(); return dst_device_to_device_stream->DoHostCallback( [input_reference, done = std::move(done), device_to_device_master_stream, dst_device_to_device_stream] { if (device_to_device_master_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)