- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for GatherVariableInfo (0.31 sec)
-
tensorflow/compiler/jit/xla_compile_on_demand_op.cc
TF_RETURN_IF_ERROR(run_result.status()); xla::ExecutionOutput execution_output = std::move(run_result).value(); absl::StatusOr<std::vector<VariableInfo>> variable_infos = 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(
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
#include "tensorflow/core/lib/core/status.h" #include "tensorflow/core/platform/thread_annotations.h" namespace tensorflow { // Creates a list of updated resource variables. absl::StatusOr<std::vector<VariableInfo>> GatherVariableInfo( OpKernelContext* ctx, const XlaCompiler::CompilationResult& compilation_result, int missing_ctx_input_prefix); // Returns pointers to inputs stored in `ctx`.
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
: nullptr); } { absl::StatusOr<std::vector<VariableInfo>> updated_variables = GatherVariableInfo(ctx, *closure.compilation_result(), closure.num_constant_args()); OP_REQUIRES_OK(ctx, updated_variables.status());
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_test.cc
inputs = {inputs.begin() + constant_input_indices.size(), inputs.end()}; { TF_ASSERT_OK_AND_ASSIGN(std::vector<VariableInfo> updated_variables, GatherVariableInfo(context_.get(), *result, constant_input_indices.size())); TF_ASSERT_OK(LockVariables(absl::MakeSpan(updated_variables))); TF_ASSERT_OK(RunPjRtExecutable(
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/xla_launch_util.cc
*ptr = new Var(write.type); return absl::OkStatus(); })); return variable; } absl::StatusOr<std::vector<VariableInfo>> GatherVariableInfo( OpKernelContext* ctx, const XlaCompiler::CompilationResult& compilation_result, int missing_ctx_input_prefix) { std::vector<VariableInfo> out;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0)