Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_task_assignment.cc58 cost_analysis_(std::move(cost_analysis)) {} in DefaultCostModel()
68 std::max(int64{1}, cost_analysis_->bytes_accessed(*instruction)); in GetParallelTaskCount()
70 cost_analysis_->flop_count(*instruction) / in GetParallelTaskCount()
90 1 * cost_analysis_->flop_count(*instruction) + in GetParallelTaskCount()
91 2 * cost_analysis_->transcendental_count(*instruction) + in GetParallelTaskCount()
92 10 * cost_analysis_->bytes_accessed(*instruction); in GetParallelTaskCount()
104 const std::unique_ptr<HloCostAnalysis> cost_analysis_; member in xla::cpu::DefaultCostModel