- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for DeviceContext (0.18 sec)
-
tensorflow/compiler/jit/pjrt_device_context.h
// Note: we currently assume the PjRtBuffer is a PjRtStreamExecutorBuffer. bool use_pjrt_tensor_buffer_; }; void PjRtDeviceToDeviceCopy(DeviceContext* send_dev_context, DeviceContext* recv_dev_context, Device* src, Device* dst, AllocatorAttributes src_alloc_attr, AllocatorAttributes dst_alloc_attr,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jul 19 19:27:39 UTC 2023 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/jit/device_context_test.cc
device_allocator_ = device_->GetAllocator(device_alloc_attr); tensorflow::DeviceContext* device_context; auto status = device_->TryGetDeviceContext(&device_context); TF_EXPECT_OK(status); device_context_.reset(device_context); } std::unique_ptr<Device> device_; tensorflow::core::RefCountPtr<DeviceContext> device_context_; tensorflow::Allocator* host_allocator_;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 3.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_host_send_device_context.h
#include "xla/stream_executor/stream.h" #include "tensorflow/core/framework/device_base.h" #include "tfrt/concurrency/async_value_ref.h" // from @tf_runtime namespace tensorflow { // XlaHostSendDeviceContext is a DeviceContext that is intended to be // used to transfer from host->device using Rendezvous. It transfers the // content of `device_memory_base` with `shape` using `stream`. Only
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 3.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_host_recv_device_context.h
#include "xla/stream_executor/stream.h" #include "tensorflow/core/framework/device_base.h" #include "tfrt/concurrency/async_value_ref.h" // from @tf_runtime namespace tensorflow { // XlaHostRecvDeviceContext is a DeviceContext that is intended to be // used to transfer from device->host using Rendezvous. It transfers the // content of `device_memory_base` with `shape` using `stream`. Only
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_device_context.h
private: // The stream executor of the device. se::StreamExecutor* stream_executor_; }; // Helper class for managing data transfers between host and XLA devices. class XlaDeviceContext : public DeviceContext { public: explicit XlaDeviceContext( std::shared_ptr<se::Stream> compute_stream, std::shared_ptr<se::Stream> host_to_device_stream, std::shared_ptr<se::Stream> device_to_host_stream,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 5.1K bytes - Viewed (0)