Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/
Dside_effect_analysis.cc74 const auto& ids = alias_analysis.GetResourceUniqueIds(operand); in FindAccessedResources()
79 const auto& ids = alias_analysis.GetResourceUniqueIds(result); in FindAccessedResources()
208 const auto& ids = alias_analysis.GetResourceUniqueIds(value); in GetResourceIdsByValue()
Dresource_alias_analysis.h55 const llvm::SmallSet<int64_t, 8>& GetResourceUniqueIds(Value resource) const;
Dresource_alias_analysis.cc505 ResourceAliasAnalysisInfo::GetResourceUniqueIds(Value resource) const { in GetResourceUniqueIds() function in mlir::TF::detail::ResourceAliasAnalysisInfo
523 for (int64_t id : GetResourceUniqueIds(resource)) { in GetResourceAliases()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dresource_device_inference.cc78 for (int64_t id : alias_analysis_.GetResourceUniqueIds(resource)) { in DeviceForResource()
98 for (int64_t id : alias_analysis_.GetResourceUniqueIds(resource)) { in AddResourceDevice()
Dtpu_cluster_formation.cc146 const auto& ids = resource_alias_analysis.GetResourceUniqueIds(operand); in CollectResourceIdsFromOp()
151 const auto& ids = resource_alias_analysis.GetResourceUniqueIds(result); in CollectResourceIdsFromOp()
187 auto ids = resource_alias_analysis.GetResourceUniqueIds(operand); in ShouldMoveOpAfterCluster()