Searched refs:resource_alias_analysis (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_cluster_formation.cc | 142 const TF::ResourceAliasAnalysis::Info& resource_alias_analysis, in CollectResourceIdsFromOp() argument
146 if (resource_alias_analysis.IsUnknownResource(operand)) continue; in CollectResourceIdsFromOp()
147 const auto& ids = resource_alias_analysis.GetResourceUniqueIds(operand); in CollectResourceIdsFromOp()
151 if (resource_alias_analysis.IsUnknownResource(result)) continue; in CollectResourceIdsFromOp()
152 const auto& ids = resource_alias_analysis.GetResourceUniqueIds(result); in CollectResourceIdsFromOp()
163 const TF::ResourceAliasAnalysis::Info& resource_alias_analysis, in ShouldMoveOpAfterCluster() argument
187 if (resource_alias_analysis.IsUnknownResource(operand)) continue; in ShouldMoveOpAfterCluster()
188 auto ids = resource_alias_analysis.GetResourceUniqueIds(operand); in ShouldMoveOpAfterCluster()
207 const TF::ResourceAliasAnalysis::Info& resource_alias_analysis) { in CollectClusterPrecedingUsers() argument
215 CollectResourceIdsFromOp(op, resource_alias_analysis, in CollectClusterPrecedingUsers()
[all …]
|
| D | tpu_dynamic_layout_pass.cc | 86 const TF::ResourceAliasAnalysis::Info& resource_alias_analysis);
101 const TF::ResourceAliasAnalysis::Info& resource_alias_analysis) { in IsSupportedInputOp() argument
107 if (resource_alias_analysis.IsUnknownResource(resource_iterator)) in IsSupportedInputOp()
110 resource_alias_analysis.GetResourceAliases(resource_iterator); in IsSupportedInputOp()
193 const TF::ResourceAliasAnalysis::Info& resource_alias_analysis) { in HandleReplicatedInputs() argument
201 if (!input_op || !IsSupportedInputOp(input_op, resource_alias_analysis)) in HandleReplicatedInputs()
230 const TF::ResourceAliasAnalysis::Info& resource_alias_analysis) { in HandleCompileAndExecutes() argument
258 maybe_replicate, resource_alias_analysis)) { in HandleCompileAndExecutes()
271 if (!IsSupportedInputOp(input_op, resource_alias_analysis)) continue; in HandleCompileAndExecutes()
289 const TF::ResourceAliasAnalysis::Info& resource_alias_analysis) { in runOnFunction() argument
[all …]
|
| D | resource_device_inference.cc | 242 const auto& resource_alias_analysis = in runOnOperation() local
250 func_op, func_op, resource_alias_analysis.GetAnalysisForFunc(func_op)); in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ |
| D | BUILD | 943 "analysis/resource_alias_analysis.cc",
947 "analysis/resource_alias_analysis.h",
|