Searched refs:alias_analysis_ (Results 1 – 10 of 10) sorted by relevance
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
D | resource_device_inference.cc | 72 : 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/ |
D | hlo_live_range.h | 104 alias_analysis_(alias_analysis), in HloLiveRange() 199 const HloAliasAnalysis& alias_analysis_; variable
|
D | hlo_live_range.cc | 35 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()
|
D | hlo_live_range_test.cc | 43 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()
|
D | buffer_assignment.h | 458 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
|
D | memory_space_assignment.h | 167 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
|
D | memory_space_assignment.cc | 196 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 …]
|
D | heap_simulator_test.cc | 305 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/ |
D | ir_emitter.h | 238 alias_analysis_.AddAliasingInformationToIrArray(hlo, array); in AddAliasingInformationToIrArray() 479 llvm_ir::AliasAnalysis alias_analysis_; variable
|
D | ir_emitter.cc | 105 alias_analysis_(hlo_module, assignment, &llvm_module->getContext()), in IrEmitter()
|