Home
last modified time | relevance | path

Searched refs:MemorySpaceAssignmentCostAnalysis (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dmemory_space_assignment.h107 class MemorySpaceAssignmentCostAnalysis {
121 virtual ~MemorySpaceAssignmentCostAnalysis() = default;
123 static StatusOr<std::unique_ptr<MemorySpaceAssignmentCostAnalysis>> Create(
210 MemorySpaceAssignmentCostAnalysis( in MemorySpaceAssignmentCostAnalysis() function
378 class MemorySpaceAssignmentCostAnalysis; variable
394 const MemorySpaceAssignmentCostAnalysis& cost_analysis,
459 const MemorySpaceAssignmentCostAnalysis& cost_analysis_;
917 const MemorySpaceAssignmentCostAnalysis& cost_analysis,
918 MemorySpaceAssignmentCostAnalysis::Cache* cache = nullptr);
1028 MemorySpaceAssignmentCostAnalysis* cost_analysis = nullptr;
H A Dmemory_space_assignment.cc245 /*static*/ StatusOr<std::unique_ptr<MemorySpaceAssignmentCostAnalysis>>
246 MemorySpaceAssignmentCostAnalysis::Create(const HloCostAnalysis& cost_analysis, in Create()
254 return absl::WrapUnique(new MemorySpaceAssignmentCostAnalysis( in Create()
259 float MemorySpaceAssignmentCostAnalysis::GetAlternateMemoryBenefit( in GetAlternateMemoryBenefit()
261 MemorySpaceAssignmentCostAnalysis::Cache* cache) const { in GetAlternateMemoryBenefit()
295 float MemorySpaceAssignmentCostAnalysis::GetMemoryBoundedness( in GetMemoryBoundedness()
297 MemorySpaceAssignmentCostAnalysis::Cache* cache) const { in GetMemoryBoundedness()
332 float MemorySpaceAssignmentCostAnalysis::GetAlternateMemoryBenefit( in GetAlternateMemoryBenefit()
334 MemorySpaceAssignmentCostAnalysis::Cache* cache) const { in GetAlternateMemoryBenefit()
344 float MemorySpaceAssignmentCostAnalysis::GetAlternateMemoryBenefit( in GetAlternateMemoryBenefit()
[all …]
H A Dmemory_space_assignment_test.cc32 using memory_space_assignment::MemorySpaceAssignmentCostAnalysis;
76 auto cost_analysis = MemorySpaceAssignmentCostAnalysis::Create( in AssignMemorySpaceUsingCostAnalysis()
344 MemorySpaceAssignmentCostAnalysis::Cache cache_;
350 : public MemorySpaceAssignmentCostAnalysis {
420 : MemorySpaceAssignmentCostAnalysis( in FakeMemorySpaceAssignmentCostAnalysis()