- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for device_context (0.16 sec)
-
tensorflow/compiler/jit/xla_device.cc
TF_ASSIGN_OR_RETURN(auto device_context, GetDeviceContextDefault()); device_context->Ref(); *out_context = device_context; return absl::OkStatus(); } // Warn about XLA_CPU/XLA_GPU exactly once. static void ShowXlaDeviceDeprecationWarning( absl::string_view compilation_device_name) { static absl::once_flag once; if (absl::StrContains(compilation_device_name, "CPU") ||
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/xla_host_send_recv_device_context_test.cc
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)); tensorflow::test::ExpectClose(origin_cpu_tensor, dest_cpu_tensor); device_context->Unref(); }
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_send_device_context.h
// done_event->Init(); // // XlaHostSendDeviceContext device_context(&stream, &gpu_dst, // shape, done_event); // device_context.CopyCPUTensorToDeviceSync( // &cpu_tensor, &device, &device_tensor); class XlaHostSendDeviceContext : public DeviceContext { public: XlaHostSendDeviceContext( se::Stream* stream, 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.7K bytes - Viewed (0) -
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_launch_util.cc
// TODO(cheshire): Avoid duplication with framework/op_kernel.h DeviceContext* device_context = nullptr; if (device != nullptr) { TF_RETURN_IF_ERROR(device->TryGetDeviceContext(&device_context)); bool using_default_context = false; auto cleanup = absl::MakeCleanup([&] { if (device_context != nullptr && !using_default_context) { device_context->Unref(); } }); if (device_context == nullptr) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
tsl::MakeConstructedAsyncValueRef<std::unique_ptr<se::Event>>( 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(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K 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_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)