Searched defs:shape_size_function (Results 1 – 3 of 3) sorted by relevance
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_t(const Shape&)> shape_size_function) in VerifiedHloModule()