Searched refs:shape_size_ (Results 1 – 3 of 3) sorted by relevance
35 : 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::SimpleCostModel58 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
241 const ShapeSizeFunction shape_size_; variable
45 : 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()