Searched refs:shape_size_function_ (Results 1 – 3 of 3) sorted by relevance
72 shape_size_function_(shape_size), in ParallelTaskAssigner()102 HloCostAnalysis::ShapeSizeFunction shape_size_function_; variable
238 shape_size_function_, module, in ComputeTargetParallelTasks()
175 : shape_size_function_(shape_size_function) {} in TargetVerifierMetadata()179 return shape_size_function_(shape); in ShapeSize()192 std::function<int64(const Shape&)> shape_size_function_;