Searched refs:variable_infos (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compile_on_demand_op.cc | 91 xla::StatusOr<std::vector<VariableInfo>> variable_infos = in Run() local
93 TF_RETURN_IF_ERROR(variable_infos.status()); in Run()
94 TF_RETURN_IF_ERROR(LockVariables(absl::MakeSpan(*variable_infos))); in Run()
97 /*missing_ctx_input_prefix=*/0, absl::MakeSpan(*variable_infos), in Run()
146 std::vector<VariableInfo> variable_infos; in Compile() local
149 inputs, variables_indices, &variable_infos)); in Compile()
151 TF_RETURN_IF_ERROR(LockVariables(absl::MakeSpan(variable_infos))); in Compile()
153 ctx, variables_indices, variable_infos, variable_args)); in Compile()
156 constant_input_indices, inputs, variable_infos, in Compile()
|
| D | get_compiler_ir.cc | 88 std::vector<VariableInfo> variable_infos; in GetCompilerIr() local
90 rmgr, dev, inputs, resource_arg_indices, &variable_infos)); in GetCompilerIr()
91 TF_RETURN_IF_ERROR(LockVariables(absl::MakeSpan(variable_infos))); in GetCompilerIr()
118 constant_arg_indices, inputs, variable_infos, dev); in GetCompilerIr()
|
| D | xla_launch_util.h | 101 absl::Span<VariableInfo const> variable_infos,
177 absl::Span<VariableInfo> variable_infos,
|
| D | xla_launch_util.cc | 179 absl::Span<VariableInfo const> variable_infos, in SnapshotResourceVariables() argument
182 Var* var = variable_infos[i].var(); in SnapshotResourceVariables()
424 absl::Span<VariableInfo> variable_infos, in PopulateOutputs() argument
531 for (int i = 0; i < variable_infos.size(); i++) { in PopulateOutputs()
532 variable_info_lookup.emplace(variable_infos[i].index(), i); in PopulateOutputs()
543 Var* var = variable_infos[variable_info_lookup[actual_input_index]].var(); in PopulateOutputs()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 168 absl::Span<VariableInfo const> variable_infos, in CompileToLocalExecutable() argument
210 constants, inputs, variable_infos, in CompileToLocalExecutable()
228 std::vector<VariableInfo> variable_infos; in Compute() local
232 inputs, resources_, &variable_infos)); in Compute()
233 OP_REQUIRES_OK(ctx, LockVariables(absl::MakeSpan(variable_infos))); in Compute()
236 variable_infos, constants_, /*lazy=*/false, in Compute()
244 resource_var_ptrs[resources_[i]] = variable_infos[i].var()->tensor(); in Compute()
294 /*missing_ctx_input_prefix=*/0, absl::MakeSpan(variable_infos), in Compute()
389 std::vector<VariableInfo> variable_infos; in Compute() local
392 inputs, resources_, &variable_infos)); in Compute()
[all …]
|