Searched refs:variable_infos (Results 1 – 1 of 1) sorted by relevance
150 std::vector<VariableInfo> variable_infos; in SnapshotResourceVariables() local152 GetVariableInfosFromCtxInputs(ctx, variable_indices, &variable_infos)); in SnapshotResourceVariables()153 TF_RETURN_IF_ERROR(LockVariables(absl::MakeSpan(variable_infos))); in SnapshotResourceVariables()156 if (variable_infos[i].var()) { in SnapshotResourceVariables()160 tensor.value = *variable_infos[i].var()->tensor(); in SnapshotResourceVariables()386 std::vector<VariableInfo> variable_infos; in PopulateOutputs() local387 variable_infos.reserve(kernel->resource_updates.size()); in PopulateOutputs()405 variable_infos.emplace_back(actual_input_index, variable); in PopulateOutputs()408 TF_RETURN_IF_ERROR(LockVariables(absl::MakeSpan(variable_infos))); in PopulateOutputs()414 if (variable_infos[i].var()->tensor()->dtype() != write.type) { in PopulateOutputs()[all …]