Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment.h80 class MemorySpaceAssignmentCostAnalysis {
88 virtual ~MemorySpaceAssignmentCostAnalysis() = default;
90 static StatusOr<std::unique_ptr<MemorySpaceAssignmentCostAnalysis>> Create(
155 MemorySpaceAssignmentCostAnalysis( in MemorySpaceAssignmentCostAnalysis() function
309 const MemorySpaceAssignmentCostAnalysis& cost_analysis,
365 const MemorySpaceAssignmentCostAnalysis& cost_analysis_;
802 const MemorySpaceAssignmentCostAnalysis& cost_analysis,
803 MemorySpaceAssignmentCostAnalysis::Cache* cache = nullptr);
Dmemory_space_assignment.cc133 /*static*/ StatusOr<std::unique_ptr<MemorySpaceAssignmentCostAnalysis>>
134 MemorySpaceAssignmentCostAnalysis::Create( in Create()
143 return absl::WrapUnique(new MemorySpaceAssignmentCostAnalysis( in Create()
149 float MemorySpaceAssignmentCostAnalysis::GetAlternateMemoryBenefit( in GetAlternateMemoryBenefit()
151 MemorySpaceAssignmentCostAnalysis::Cache* cache) const { in GetAlternateMemoryBenefit()
185 float MemorySpaceAssignmentCostAnalysis::GetMemoryBoundedness( in GetMemoryBoundedness()
187 MemorySpaceAssignmentCostAnalysis::Cache* cache) const { in GetMemoryBoundedness()
231 int MemorySpaceAssignmentCostAnalysis::CalculateComputationNestLevel( in CalculateComputationNestLevel()
248 float MemorySpaceAssignmentCostAnalysis::GetInstructionElapsedDueToCompute( in GetInstructionElapsedDueToCompute()
257 float MemorySpaceAssignmentCostAnalysis::GetInstructionElapsedDueToMemory( in GetInstructionElapsedDueToMemory()
[all …]
Dmemory_space_assignment_test.cc56 auto cost_analysis = MemorySpaceAssignmentCostAnalysis::Create( in AssignMemorySpaceUsingCostAnalysis()
312 MemorySpaceAssignmentCostAnalysis::Cache cache_;
318 : public MemorySpaceAssignmentCostAnalysis {
388 : MemorySpaceAssignmentCostAnalysis( in FakeMemorySpaceAssignmentCostAnalysis()