| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/ |
| D | side_effect_analysis.h | 64 alias_analysis_(alias_analysis) { in SideEffectAnalysisInfo()
73 alias_analysis_(alias_analysis) { in SideEffectAnalysisInfo()
109 return alias_analysis_.IsUniqueResourceAllocationId(resource_id); in IsUniqueResourceAllocationId()
183 const TF::ResourceAliasAnalysis::Info& alias_analysis_; variable
204 ResourceAliasAnalysis alias_analysis_;
|
| D | side_effect_analysis.cc | 567 &child_region, op_side_effect_collector_, alias_analysis_); in AnalyzeRegion()
610 alias_analysis_); in AnalyzeOp()
732 : alias_analysis_(module) { in SideEffectAnalysis()
740 alias_analysis_.GetAnalysisForFunc(func)); in SideEffectAnalysis()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | resource_device_inference.cc | 74 : alias_analysis_(alias_analysis) {} in PerFunctionResult()
79 if (alias_analysis_.IsUnknownResource(resource)) return llvm::None; in DeviceForResource()
80 for (int64_t id : alias_analysis_.GetResourceUniqueIds(resource)) { in DeviceForResource()
99 if (alias_analysis_.IsUnknownResource(resource)) return success(); in AddResourceDevice()
100 for (int64_t id : alias_analysis_.GetResourceUniqueIds(resource)) { in AddResourceDevice()
114 const TF::ResourceAliasAnalysis::Info& alias_analysis_; member in mlir::TF::__anonbeac22d10111::PerFunctionResult
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_live_range.cc | 49 const HloBuffer& buffer = alias_analysis_.GetBufferContainingValue(value); in NormalizeAliasedBuffers()
215 alias_analysis_.dataflow_analysis().GetInstructionValueSet( in CalculateBufferStartEndMap()
248 for (const HloValue* value : alias_analysis_.dataflow_analysis().values()) { in ComputePeakMemoryMoment()
291 for (const HloValue* value : alias_analysis_.dataflow_analysis().values()) { in ToString()
304 for (const HloValue* value : alias_analysis_.dataflow_analysis().values()) { in ToString()
|
| D | hlo_live_range.h | 113 alias_analysis_(alias_analysis), in HloLiveRange()
220 const HloAliasAnalysis& alias_analysis_; variable
|
| 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::__anon099ea49a0111::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()
510 alias_analysis_(std::move(alias_analysis)), in BufferAssignment()
579 std::unique_ptr<HloAliasAnalysis> alias_analysis_; variable
|
| D | memory_space_assignment.cc | 301 for (const HloBuffer* buffer : alias_analysis_->ComputeBuffersAt( in GetMemoryBoundedness()
979 alias_analysis_(alias_analysis), in AlternateMemoryBestFitHeap()
1183 &alias_analysis_.dataflow_analysis().GetUniqueValueAt( in IsUseAllowedInAlternateMemory()
1216 &alias_analysis_.dataflow_analysis().GetUniqueValueAt( in IsUseAllowedInAlternateMemory()
1242 &alias_analysis_.dataflow_analysis().GetUniqueValueAt( in IsUseAllowedInAlternateMemory()
1307 alias_analysis_.GetBufferContainingValue(*interval.buffer); in AppendBufferInfoDebugString()
1350 const HloBuffer& buffer = alias_analysis_.GetUniqueBufferAt(instruction); in AppendScopedAllocationBufferInfoDebugString()
1381 alias_analysis_.GetUniqueBufferAt(position.instruction, position.index); in AppendAllocationInfoDebugString()
1429 alias_analysis_, hlo_live_range_, options_); in Finish()
2331 &alias_analysis_.dataflow_analysis().GetUniqueValueAt( in AliasedRequiredAssignmentForUse()
[all …]
|
| D | memory_space_assignment.h | 217 alias_analysis_(std::move(alias_analysis)), in MemorySpaceAssignmentCostAnalysis()
224 std::unique_ptr<HloAliasAnalysis> alias_analysis_; variable
1261 const HloAliasAnalysis& alias_analysis() { return alias_analysis_; } in alias_analysis()
1596 const HloAliasAnalysis& alias_analysis_; variable
|
| D | heap_simulator_test.cc | 305 alias_analysis_ = HloAliasAnalysis::Run(module_.get()).value(); 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::__anoneda8e6790111::HeapSimulatorTracker
|
| /external/pytorch/torch/csrc/jit/runtime/ |
| D | operator.h | 70 mutable std::optional<c10::AliasAnalysisKind> alias_analysis_; member
177 if (unmaterializedSchema.alias_analysis_.has_value()) { in schema()
180 *unmaterializedSchema.alias_analysis_); in schema()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.h | 247 alias_analysis_.AddAliasingInformationToIrArray(hlo, array); in AddAliasingInformationToIrArray()
538 llvm_ir::AliasAnalysis alias_analysis_; variable
|
| D | ir_emitter.cc | 116 alias_analysis_(hlo_module, assignment, &llvm_module->getContext()), in IrEmitter()
|