- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for op_device_context (0.24 sec)
-
tensorflow/compiler/jit/xla_compile_on_demand_op.cc
OpKernelContext* ctx) { xla::LocalClient* client = static_cast<xla::LocalClient*>(xla_device_compiler->client()); se::Stream* stream = ctx->op_device_context() ? ctx->op_device_context()->stream() : nullptr; std::shared_ptr<se::DeviceMemoryAllocator> allocator_ptr = GetAllocator(ctx->device(), stream, platform_info_); se::DeviceMemoryAllocator* allocator = allocator_ptr.get();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 08:39:39 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util.cc
// stream now otherwise we will create a race condition. auto* gpu_device_context = static_cast<GPUDeviceContext*>(ctx->op_device_context()); TF_RETURN_IF_ERROR(gpu_device_context->stream()->WaitFor( gpu_device_context->host_to_device_stream())); } } else { // No copy required. ctx->set_output(output_num, const_tensor);
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/c/kernels.cc
#else auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); if (cc_ctx->op_device_context() == nullptr) { // CPU Device status->status = absl::FailedPreconditionError( "Accessing device stream is not supported for a CPU device."); return nullptr; } else if (!cc_ctx->op_device_context()->IsPluggableDevice()) { status->status = absl::FailedPreconditionError(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
using PjRtExecutableClosureStore = ExecutableClosureStore<xla::PjRtLoadedExecutable, xla::PjRtClient>; se::Stream* GetStream(OpKernelContext* ctx) { return ctx->op_device_context() ? ctx->op_device_context()->stream() : nullptr; } XlaComputationLaunchContext GetLaunchContext( const XlaPlatformInfo& platform_info, OpKernelContext* ctx,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0)