- Sort Score
- Result 10 results
- Languages All
Results 11 - 14 of 14 for device_context (2.07 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/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_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)