Searched refs:IsSimpleLayout (Results 1 – 1 of 1) sorted by relevance
932 bool IsSimpleLayout(const Layout& layout) { in IsSimpleLayout() function941 CHECK(!lhs_shape.has_layout() || IsSimpleLayout(lhs_shape.layout())) in AreGemmShapes()943 CHECK(!rhs_shape.has_layout() || IsSimpleLayout(rhs_shape.layout())) in AreGemmShapes()945 CHECK(!output_shape.has_layout() || IsSimpleLayout(output_shape.layout())) in AreGemmShapes()