Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dresource_device_inference.cc72 : alias_analysis_(alias_analysis) {} in PerFunctionResult()
77 if (alias_analysis_.IsUnknownResource(resource)) return llvm::None; in DeviceForResource()
78 for (int64_t id : alias_analysis_.GetResourceUniqueIds(resource)) { in DeviceForResource()
97 if (alias_analysis_.IsUnknownResource(resource)) return success(); in AddResourceDevice()
98 for (int64_t id : alias_analysis_.GetResourceUniqueIds(resource)) { in AddResourceDevice()
112 const TF::ResourceAliasAnalysis::Info& alias_analysis_; member in mlir::TF::__anone6a4cbe20111::PerFunctionResult
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_live_range.h104 alias_analysis_(alias_analysis), in HloLiveRange()
199 const HloAliasAnalysis& alias_analysis_; variable
Dhlo_live_range.cc35 for (const HloBuffer& hlo_buffer : alias_analysis_.buffers()) { in NormalizeAliasedBuffers()
131 for (const HloValue* value : alias_analysis_.dataflow_analysis().values()) { in CalculateBufferStartEndMap()
222 for (const HloValue* value : alias_analysis_.dataflow_analysis().values()) { in ToString()
Dhlo_live_range_test.cc43 alias_analysis_ = HloAliasAnalysis::Run(module_.get()).ValueOrDie(); in Analyze()
44 hlo_live_range_ = HloLiveRange::Run(schedule, *alias_analysis_, in Analyze()
51 std::unique_ptr<HloAliasAnalysis> alias_analysis_; member in xla::__anon60e75d0b0111::HloLiveRangeTest
59 return &alias_analysis_->dataflow_analysis().GetUniqueValueAt(instruction, in BufferAt()
Dbuffer_assignment.h458 return alias_analysis_->dataflow_analysis(); in dataflow_analysis()
461 HloAliasAnalysis& alias_analysis() const { return *alias_analysis_; } in alias_analysis()
507 alias_analysis_(std::move(alias_analysis)), in BufferAssignment()
575 std::unique_ptr<HloAliasAnalysis> alias_analysis_; variable
Dmemory_space_assignment.h167 alias_analysis_(std::move(alias_analysis)), in MemorySpaceAssignmentCostAnalysis()
175 std::unique_ptr<HloAliasAnalysis> alias_analysis_; variable
955 alias_analysis_(alias_analysis), in AlternateMemoryBestFitHeap()
1000 const HloAliasAnalysis& alias_analysis() { return alias_analysis_; } in alias_analysis()
1324 const HloAliasAnalysis& alias_analysis_; variable
Dmemory_space_assignment.cc196 for (const HloBuffer* buffer : alias_analysis_->ComputeBuffersAt( in GetMemoryBoundedness()
900 &alias_analysis_.dataflow_analysis().GetUniqueValueAt( in IsUseAllowedInAlternateMemory()
933 &alias_analysis_.dataflow_analysis().GetUniqueValueAt( in IsUseAllowedInAlternateMemory()
959 &alias_analysis_.dataflow_analysis().GetUniqueValueAt( in IsUseAllowedInAlternateMemory()
1018 alias_analysis_.GetBufferContainingValue(*interval.buffer); in AppendBufferInfoDebugString()
1069 alias_analysis_.GetBufferContainingValue(*value.value()); in AppendAllocationInfoDebugString()
1090 alias_analysis_, hlo_live_range_, options_); in Finish()
1657 &alias_analysis_.dataflow_analysis().GetUniqueValueAt( in AliasedRequiredAssignmentForUse()
1714 &alias_analysis_.dataflow_analysis().GetUniqueValueAt(instruction, index); in AddRequiredAssignment()
1724 const HloModule& module = alias_analysis_.dataflow_analysis().module(); in AddInputAndOutputRequiredAssignments()
[all …]
Dheap_simulator_test.cc305 alias_analysis_ = HloAliasAnalysis::Run(module_.get()).ConsumeValueOrDie(); in RunWholeModule()
326 *alias_analysis_, size_fn) in RunWholeModule()
335 return &alias_analysis_->dataflow_analysis().GetUniqueValueAt(instruction, in BufferAt()
388 alias_analysis_ = in Init()
396 *alias_analysis_, zero_size, options) in Init()
401 std::unique_ptr<HloAliasAnalysis> alias_analysis_; member in xla::__anon44f19eea0111::HeapSimulatorTracker
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h238 alias_analysis_.AddAliasingInformationToIrArray(hlo, array); in AddAliasingInformationToIrArray()
479 llvm_ir::AliasAnalysis alias_analysis_; variable
Dir_emitter.cc105 alias_analysis_(hlo_module, assignment, &llvm_module->getContext()), in IrEmitter()