experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Skip shared variables for VariableAccessor::GetConstDeclarations().

Browse Source 
PiperOrigin-RevId: 263191156
This commit is contained in:
Juhyun Lee 2019-08-13 12:30:09 -07:00 committed by TensorFlower Gardener
parent 8cd1d56995
commit dadbec4cab
1 changed files with 7 additions and 1 deletions

8
tensorflow/lite/delegates/gpu/gl/compiler/variable_accessor.cc
View File

@ -371,10 +371,16 @@ std::string VariableAccessor::GetConstDeclarations() const {
// with index.
std::string declarations;
for (const auto& variable : name_to_variable_) {
// Skip shared variables.
const std::string& variable_name = variable.second.name;
if (shared_variables_.find(variable_name) != shared_variables_.end()) {
continue;
}
const auto& value = variable.second.value;
if (IsVariableLength(value)) {
absl::StrAppend(&declarations, "const ", GetVariableType(value), " ",
variable.second.name, "[] = ");
variable_name, "[] = ");
GetValue(value, &declarations);
absl::StrAppend(&declarations, ";\n");
}