Searched defs:shape_size_function (Results 1 – 4 of 4) sorted by relevance
33 std::function<int64(const Shape&)> shape_size_function) in ShapeVerifier()211 TargetVerifierMetadata(std::function<int64(const Shape&)> shape_size_function) in TargetVerifierMetadata()240 std::function<int64(const Shape&)> shape_size_function) in DefaultVerifierMetadata()
36 : shape_size_function_(std::move(shape_size_function)), in shape_size_function_()
45 auto shape_size_function = [&](const Shape& shape) { in TEST_F() local
38 std::function<int64(const Shape&)> shape_size_function) in VerifiedHloModule()