Searched refs:cost_analysis_ (Results 1 – 3 of 3) sorted by relevance
59 cost_analysis_(std::move(cost_analysis)) {} in DefaultCostModel()69 std::max(int64{1}, cost_analysis_->bytes_accessed(*instruction)); in GetParallelTaskCount()71 cost_analysis_->flop_count(*instruction) / in GetParallelTaskCount()92 1 * cost_analysis_->flop_count(*instruction) + in GetParallelTaskCount()93 2 * cost_analysis_->transcendental_count(*instruction) + in GetParallelTaskCount()94 10 * cost_analysis_->bytes_accessed(*instruction); in GetParallelTaskCount()106 const std::unique_ptr<HloCostAnalysis> cost_analysis_; member in xla::cpu::DefaultCostModel
251 cost_analysis_.flop_count(instruction) / in GetInstructionElapsedDueToCompute()252 cost_analysis_.per_second_rate(HloCostAnalysis::kFlopsKey), in GetInstructionElapsedDueToCompute()253 cost_analysis_.transcendental_count(instruction) / in GetInstructionElapsedDueToCompute()254 cost_analysis_.per_second_rate(HloCostAnalysis::kTranscendentalsKey)); in GetInstructionElapsedDueToCompute()261 float bytes_accessed = cost_analysis_.bytes_accessed(instruction); in GetInstructionElapsedDueToMemory()264 cost_analysis_.per_second_rate(HloCostAnalysis::kBytesAccessedKey); in GetInstructionElapsedDueToMemory()268 float operand_bytes_accessed = cost_analysis_.operand_bytes_accessed( in GetInstructionElapsedDueToMemory()276 cost_analysis_.per_second_rate(HloCostAnalysis::kBytesAccessedKey); in GetInstructionElapsedDueToMemory()282 cost_analysis_.output_bytes_accessed(instruction); in GetInstructionElapsedDueToMemory()289 cost_analysis_.per_second_rate(HloCostAnalysis::kBytesAccessedKey); in GetInstructionElapsedDueToMemory()[all …]
95 const HloCostAnalysis& cost_analysis() const { return cost_analysis_; } in cost_analysis()162 : cost_analysis_(cost_analysis), in MemorySpaceAssignmentCostAnalysis()172 const HloCostAnalysis& cost_analysis_;365 const MemorySpaceAssignmentCostAnalysis& cost_analysis_; variable