- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for DeviceContext (0.63 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/xla_device.h
Status Sync() override; Status TryGetDeviceContext(DeviceContext** out_context) override TF_LOCKS_EXCLUDED(mu_); Status MakeTensorFromProto(const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, Tensor* tensor) override TF_LOCKS_EXCLUDED(mu_); Status MakeTensorFromProto(DeviceContext* device_context,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 13.4K 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.cc
} return device_contexts_; } absl::StatusOr<DeviceContext*> XlaDevice::GetDeviceContextWithIndex(int index) { mutex_lock lock(mu_); TF_ASSIGN_OR_RETURN(auto device_contexts, GetDeviceContextLocked()); return device_contexts.at(index); } absl::StatusOr<DeviceContext*> XlaDevice::GetDeviceContextDefault() { return GetDeviceContextWithIndex(0); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 21:05:42 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_device_context.cc
tensorflow::AsyncValueTensor::FromTensor(output_tensor); result_tensor->GetBuffer()->GetReadyFuture().OnReady(std::move(done)); } 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: Sat Apr 13 08:49:31 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_gpu_test.cc
SetDevice(device_type, std::move(device)); XlaShapeLayoutHelpers::ShapeDeterminationFns shape_fns{ UseNoPreferenceLayoutFn(), IdentityShapeRepresentationFn()}; device_context_ = core::RefCountPtr<DeviceContext>( new PjRtDeviceContext(shape_fns, /*use_pjrt_tensor_buffer=*/true)); // Get the host allocator. AllocatorAttributes host_alloc_attr; host_alloc_attr.set_on_host(true);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 10K 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) -
tensorflow/compiler/jit/xla_tpu_device.cc
// strictly hierarchical, that is, transfers between devices on different hosts // can only take place using the host network. void TpuDeviceToDeviceCopy(DeviceContext* src_dev_context, DeviceContext* dst_dev_context, Device* src, Device* dst, AllocatorAttributes src_allocator_attrs, AllocatorAttributes dst_allocator_attrs,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 20.9K bytes - Viewed (0)