Searched refs:shape_size_function (Results 1 – 3 of 3) sorted by relevance
46 auto shape_size_function = [&](const Shape& shape) { in TEST_F() local54 HloCostAnalysis cost_analysis(shape_size_function); in TEST_F()
174 TargetVerifierMetadata(std::function<int64(const Shape&)> shape_size_function) in TargetVerifierMetadata() argument175 : shape_size_function_(shape_size_function) {} in TargetVerifierMetadata()201 std::function<int64(const Shape&)> shape_size_function) in DefaultVerifierMetadata() argument202 : TargetVerifierMetadata(shape_size_function), in DefaultVerifierMetadata()
50 std::function<int64(const Shape&)> shape_size_function) in VerifiedHloModule() argument54 /*instruction_can_change_layout_func=*/{}, shape_size_function) {} in VerifiedHloModule()