Searched defs:shape_size_function (Results 1 – 3 of 3) sorted by relevance
174 TargetVerifierMetadata(std::function<int64(const Shape&)> shape_size_function) in TargetVerifierMetadata()201 std::function<int64(const Shape&)> shape_size_function) in DefaultVerifierMetadata()
46 auto shape_size_function = [&](const Shape& shape) { in TEST_F() local
50 std::function<int64(const Shape&)> shape_size_function) in VerifiedHloModule()