Searched refs:alias_analysis (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | copy_insertion.cc | 256 Status AddCopiesForWhile(const HloAliasAnalysis& alias_analysis, in AddCopiesForWhile() argument
262 if (!IndicesToCopyForWhile(alias_analysis.dataflow_analysis(), xla_while, in AddCopiesForWhile()
317 Status AddCopiesForConditional(const HloAliasAnalysis& alias_analysis, in AddCopiesForConditional() argument
475 CopyRemover(const HloModule& module, const HloAliasAnalysis& alias_analysis, in CopyRemover() argument
477 : dataflow_(alias_analysis.dataflow_analysis()), ordering_(ordering) { in CopyRemover()
482 for (const HloBuffer& buffer : alias_analysis.buffers()) { in CopyRemover()
966 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in AddCopiesToResolveInterference()
972 TF_RETURN_IF_ERROR(AddCopiesForWhile(*alias_analysis, instruction)); in AddCopiesToResolveInterference()
975 AddCopiesForConditional(*alias_analysis, instruction)); in AddCopiesToResolveInterference()
991 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in AddSpecialCaseCopies()
[all …]
|
| D | hlo_alias_analysis.cc | 496 auto alias_analysis = absl::WrapUnique(new HloAliasAnalysis(module)); in Run() local
497 TF_ASSIGN_OR_RETURN(alias_analysis->dataflow_analysis_, in Run()
502 BufferValueMap buffer_map(module, alias_analysis->dataflow_analysis()); in Run()
510 alias_analysis->buffers_.reserve(sorted_buffer_numbers.size()); in Run()
517 alias_analysis->buffers_.emplace_back(next_id++, sorted_values); in Run()
519 alias_analysis->value_to_buffer_[value] = in Run()
520 &alias_analysis->buffers_.back(); in Run()
524 TF_DCHECK_OK(alias_analysis->Verify()); in Run()
526 XLA_VLOG_LINES(2, alias_analysis->ToString()); in Run()
527 return std::move(alias_analysis); in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | BUILD | 30 name = "alias_analysis",
31 srcs = ["alias_analysis.cc"],
32 hdrs = ["alias_analysis.h"],
51 ":alias_analysis",
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | BUILD | 93 "//tensorflow/compiler/xla/service/llvm_ir:alias_analysis",
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | BUILD | 205 "//tensorflow/compiler/xla/service/llvm_ir:alias_analysis",
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | BUILD | 290 "//tensorflow/compiler/xla/service/llvm_ir:alias_analysis",
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | BUILD | 131 "//tensorflow/compiler/xla/service/llvm_ir:alias_analysis",
|