Searched refs:IsSimpleLayout (Results 1 – 1 of 1) sorted by relevance
745 bool IsSimpleLayout(const Layout& layout) { in IsSimpleLayout() function754 CHECK(!lhs_shape.has_layout() || IsSimpleLayout(lhs_shape.layout())) in AreGemmShapes()756 CHECK(!rhs_shape.has_layout() || IsSimpleLayout(rhs_shape.layout())) in AreGemmShapes()758 CHECK(!output_shape.has_layout() || IsSimpleLayout(output_shape.layout())) in AreGemmShapes()