Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dxla_compile_on_demand_op.cc94 TF_RETURN_IF_ERROR(LockVariables(absl::MakeSpan(*variable_infos))); in Run()
151 TF_RETURN_IF_ERROR(LockVariables(absl::MakeSpan(variable_infos))); in Compile()
Dxla_launch_util.h109 Status LockVariables(absl::Span<VariableInfo> variables)
Dget_compiler_ir.cc91 TF_RETURN_IF_ERROR(LockVariables(absl::MakeSpan(variable_infos))); in GetCompilerIr()
Dxla_launch_util.cc132 Status LockVariables(absl::Span<VariableInfo> variables) { in LockVariables() function
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc233 OP_REQUIRES_OK(ctx, LockVariables(absl::MakeSpan(variable_infos))); in Compute()
393 OP_REQUIRES_OK(ctx, LockVariables(absl::MakeSpan(variable_infos))); in Compute()
538 OP_REQUIRES_OK(ctx, LockVariables(absl::MakeSpan(*variable_infos))); in Compute()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_execute_op.cc272 TF_RETURN_IF_ERROR(LockVariables(absl::MakeSpan(variables))); in BuildComputationInputs()