Searched refs:shape_checker (Results 1 – 1 of 1) sorted by relevance
226 auto shape_checker = [&](const xla::Shape& pshape, in InputShapeMatches() local251 shape_checker) in InputShapeMatches()