- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for device_context (2.73 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_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/xla_device.h
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/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/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_launch_util_test.cc
device_type, xla::GetTfrtCpuClient(/*asynchronous=*/true, /*cpu_device_count=*/1) .value())); // device_context_ should be a PjRtDeviceContext. TF_CHECK_OK(device_->TryGetDeviceContext(&device_context_)); // 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 Feb 21 09:53:30 UTC 2024 - 28.8K 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)