Home
last modified time | relevance | path

Searched refs:LockVariables (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_launch_util.h116 Status LockVariables(absl::Span<VariableInfo*> variables)
118 Status LockVariables(absl::Span<VariableInfo> variables)
H A Dxla_compile_on_demand_op.cc102 TF_RETURN_IF_ERROR(LockVariables(absl::MakeSpan(*variable_infos))); in Run()
157 TF_RETURN_IF_ERROR(LockVariables(absl::MakeSpan(variable_infos))); in Compile()
H A Dxla_launch_util.cc146 Status LockVariables(absl::Span<VariableInfo*> variables) { in LockVariables() function
191 Status LockVariables(absl::Span<VariableInfo> variables) { in LockVariables() function
197 return LockVariables(absl::MakeSpan(variable_ptrs)); in LockVariables()
H A Dget_compiler_ir.cc125 TF_RETURN_IF_ERROR(LockVariables(absl::MakeSpan(variable_infos))); in GetCompilerIr()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/kernels/
H A Dxla_ops.cc212 TF_RETURN_IF_ERROR(LockVariables(absl::MakeSpan(variable_infos))); in GetXlaCompilerArgsAndSnapshotVariables()
384 OP_REQUIRES_OK_ASYNC(ctx, LockVariables(absl::MakeSpan(variable_infos)), in ComputeAsync()
630 OP_REQUIRES_OK(ctx, LockVariables(absl::MakeSpan(*variable_infos))); in Compute()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_reshard_variables_op.cc181 TF_RETURN_IF_ERROR(LockVariables(absl::MakeSpan(variables))); in DoTpuExecute()
H A Dtpu_execute_op.cc271 TF_RETURN_IF_ERROR(LockVariables(absl::MakeSpan(variables))); in BuildComputationInputs()