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 | 141 const TF::ResourceAliasAnalysis::Info& resource_alias_analysis, in CollectResourceIdsFromOp() argument
145 if (resource_alias_analysis.IsUnknownResource(operand)) continue; in CollectResourceIdsFromOp()
146 const auto& ids = resource_alias_analysis.GetResourceUniqueIds(operand); in CollectResourceIdsFromOp()
150 if (resource_alias_analysis.IsUnknownResource(result)) continue; in CollectResourceIdsFromOp()
151 const auto& ids = resource_alias_analysis.GetResourceUniqueIds(result); in CollectResourceIdsFromOp()
162 const TF::ResourceAliasAnalysis::Info& resource_alias_analysis, in ShouldMoveOpAfterCluster() argument
186 if (resource_alias_analysis.IsUnknownResource(operand)) continue; in ShouldMoveOpAfterCluster()
187 auto ids = resource_alias_analysis.GetResourceUniqueIds(operand); in ShouldMoveOpAfterCluster()
206 const TF::ResourceAliasAnalysis::Info& resource_alias_analysis) { in CollectClusterPrecedingUsers() argument
214 CollectResourceIdsFromOp(op, resource_alias_analysis, in CollectClusterPrecedingUsers()
[all …]
|
| D | tpu_dynamic_layout_pass.cc | 86 const TF::ResourceAliasAnalysis::Info& resource_alias_analysis);
94 const TF::ResourceAliasAnalysis::Info& resource_alias_analysis) { in IsSupportedInputOp() argument
100 if (resource_alias_analysis.IsUnknownResource(resource_iterator)) in IsSupportedInputOp()
103 resource_alias_analysis.GetResourceAliases(resource_iterator); in IsSupportedInputOp()
186 const TF::ResourceAliasAnalysis::Info& resource_alias_analysis) { in HandleReplicatedInputs() argument
194 if (!input_op || !IsSupportedInputOp(input_op, resource_alias_analysis)) in HandleReplicatedInputs()
223 const TF::ResourceAliasAnalysis::Info& resource_alias_analysis) { in HandleCompileAndExecutes() argument
251 maybe_replicate, resource_alias_analysis)) { in HandleCompileAndExecutes()
264 if (!IsSupportedInputOp(input_op, resource_alias_analysis)) continue; in HandleCompileAndExecutes()
282 const TF::ResourceAliasAnalysis::Info& resource_alias_analysis) { in runOnFunction() argument
[all …]
|
| D | resource_device_inference.cc | 235 const auto& resource_alias_analysis = in runOnOperation() local
243 func_op, func_op, resource_alias_analysis.GetAnalysisForFunc(func_op)); in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ |
| D | BUILD | 821 "analysis/resource_alias_analysis.cc",
825 "analysis/resource_alias_analysis.h",
|