Searched refs:shape_size_ (Results 1 – 3 of 3) sorted by relevance
34 : max_parallelism_(max_parallelism), shape_size_(shape_size) {} in SimpleCostModel()39 const int64 instruction_cost = shape_size_(instruction->shape()); in GetParallelTaskCount()48 const HloCostAnalysis::ShapeSizeFunction shape_size_; member in xla::cpu::SimpleCostModel57 shape_size_(shape_size), in DefaultCostModel()80 instruction_cost = shape_size_(instruction->shape()); in GetParallelTaskCount()103 const HloCostAnalysis::ShapeSizeFunction shape_size_; member in xla::cpu::DefaultCostModel
40 : shape_size_(shape_size), per_second_rates_(per_second_rates) {} in HloCostAnalysis()129 return shape_size_(shape); in GetShapeSize()779 HloCostAnalysis visitor(shape_size_, per_second_rates_); in ProcessNestedSubcomputation()786 HloCostAnalysis visitor(shape_size_, per_second_rates_); in ProcessUnnestedSubcomputation()
202 const ShapeSizeFunction shape_size_; variable