Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_merge_variables_with_execute.cc96 llvm::SmallDenseMap<Value, VariableAccessInfo, 8> per_resource_info; member
174 infos.per_resource_info.try_emplace(resource, VariableAccessInfo()); in BuildVariableAccessInfo()
179 infos.per_resource_info.shrink_and_clear(); in BuildVariableAccessInfo()
216 auto info_it = infos.per_resource_info.find(read.resource()); in BuildVariableAccessInfo()
217 if (info_it == infos.per_resource_info.end()) continue; in BuildVariableAccessInfo()
220 infos.per_resource_info.erase(info_it); in BuildVariableAccessInfo()
225 return infos.per_resource_info.count(resource) == 0; in BuildVariableAccessInfo()
230 if (infos.per_resource_info.empty()) { in BuildVariableAccessInfo()
253 auto it = infos.per_resource_info.find(resource); in BuildVariableAccessInfo()
254 if (it == infos.per_resource_info.end()) continue; in BuildVariableAccessInfo()
[all …]