Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_task_assignment.cc35 : max_parallelism_(max_parallelism), shape_size_(shape_size) {} in SimpleCostModel()
40 const int64 instruction_cost = shape_size_(instruction->shape()); in GetParallelTaskCount()
49 const HloCostAnalysis::ShapeSizeFunction shape_size_; member in xla::cpu::SimpleCostModel
58 shape_size_(shape_size), in DefaultCostModel()
82 instruction_cost = shape_size_(instruction->shape()); in GetParallelTaskCount()
105 const HloCostAnalysis::ShapeSizeFunction shape_size_; member in xla::cpu::DefaultCostModel
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h241 const ShapeSizeFunction shape_size_; variable
Dhlo_cost_analysis.cc45 : shape_size_(shape_size), per_second_rates_(per_second_rates) {} in HloCostAnalysis()
142 return shape_size_(shape); in GetShapeSize()
1138 auto visitor = CreateNestedCostAnalysis(shape_size_, per_second_rates_); in ProcessSubcomputation()