Searched refs:isRowMajorBatchMatmul (Results 1 – 4 of 4) sorted by relevance
191 TEST(isRowMajorBatchMatmul, Simple) { in TEST() argument201 EXPECT_THAT(maps, Truly(isRowMajorBatchMatmul)); in TEST()204 TEST(isRowMajorBatchMatmul, BindingShifted) { in TEST() argument214 EXPECT_THAT(maps, Truly(isRowMajorBatchMatmul)); in TEST()217 TEST(isRowMajorBatchMatmul, BindingSwapped) { in TEST() argument227 EXPECT_THAT(maps, Truly(isRowMajorBatchMatmul)); in TEST()230 TEST(isRowMajorBatchMatmul, FirstInputSwapped) { in TEST() argument240 EXPECT_THAT(maps, Not(Truly(isRowMajorBatchMatmul))); in TEST()
67 bool mlir::isRowMajorBatchMatmul(ArrayAttr indexingMaps) { in isRowMajorBatchMatmul() function in mlir
46 bool isRowMajorBatchMatmul(ArrayAttr indexingMaps);
83 /*methodName=*/"isRowMajorBatchMatmul",86 return mlir::isRowMajorBatchMatmul($_op.getIndexingMaps());