- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for XlaHostRecvDeviceContext (0.39 sec)
-
tensorflow/compiler/jit/xla_host_recv_device_context.h
// done_event->Init(); // Tensor dest_cpu_tensor; // // XlaHostRecvDeviceContext device_context(&stream, gpu_dst, // shape, done_event); // device_context.CopyDeviceTensorToCPUSync( // &device_tensor, "", &device, &dest_cpu_tensor); class XlaHostRecvDeviceContext : public DeviceContext { public: XlaHostRecvDeviceContext( se::Stream* stream, const se::DeviceMemoryBase& device_memory_base,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_host_send_recv_device_context_test.cc
tsl::AsyncValueRef<std::unique_ptr<se::Event>> done_event = tsl::MakeConstructedAsyncValueRef<std::unique_ptr<se::Event>>( std::move(se_event)); XlaHostRecvDeviceContext* device_context = new XlaHostRecvDeviceContext(stream.get(), gpu_dst, shape, done_event); TF_ASSERT_OK(device_context->CopyDeviceTensorToCPUSync( &device_tensor, "", device_.get(), &dest_cpu_tensor));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_host_recv_device_context.cc
#include "tensorflow/compiler/jit/xla_host_recv_device_context.h" #include "tensorflow/compiler/tf2xla/shape_util.h" #include "tensorflow/compiler/tf2xla/type_util.h" namespace tensorflow { void XlaHostRecvDeviceContext::CopyDeviceTensorToCPU( const Tensor* device_tensor, StringPiece tensor_name, Device* device, Tensor* cpu_tensor, StatusCallback done) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
std::move(event)); Rendezvous::Args args; // Rendezvous::Args owns the device context pointer. args.device_context = new XlaHostRecvDeviceContext( stream, device_memory_base, shape, done_event); Tensor host_tensor; TF_RETURN_IF_ERROR( ctx->rendezvous()->Send(parsed_key, args, host_tensor, false));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0)