- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for PopulateOutputs (0.28 sec)
-
tensorflow/compiler/jit/xla_compile_on_demand_op.cc
GatherVariableInfo(ctx, *result, 0); TF_RETURN_IF_ERROR(variable_infos.status()); TF_RETURN_IF_ERROR(LockVariables(absl::MakeSpan(*variable_infos))); TF_RETURN_IF_ERROR(launch_context.PopulateOutputs( ctx, result, execution_output.ConsumeResult(), /*missing_ctx_input_prefix=*/0, absl::MakeSpan(*variable_infos), input_output_alias, snapshot_ptrs)); return absl::OkStatus(); }
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.h
// // // Assumes that the first `missing_ctx_input_prefix` inputs to the // compilation_result are missing and adjusts input indices accordingly. Status PopulateOutputs( OpKernelContext* ctx, const XlaCompiler::CompilationResult* compilation_result, xla::ScopedShapedBuffer output, int missing_ctx_input_prefix, absl::Span<VariableInfo> variable_infos,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
OP_REQUIRES_ASYNC(ctx, execution_output.ok(), execution_output.status(), done); OP_REQUIRES_OK_ASYNC( ctx, launch_context.PopulateOutputs( ctx, compilation_result, execution_output->ConsumeResult(), /*missing_ctx_input_prefix=*/0, absl::MakeSpan(variable_infos), input_output_alias, resource_var_ptrs),
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.cc
out.emplace_back(actual_input_index, handle.name(), variable, handle.definition_stack_trace()); } return std::move(out); } Status XlaComputationLaunchContext::PopulateOutputs( OpKernelContext* ctx, const XlaCompiler::CompilationResult* compilation_result, ScopedShapedBuffer output, int missing_ctx_input_prefix, absl::Span<VariableInfo> variable_infos,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0)