Searched defs:isRowMajorBatchMatmul (Results 1 – 2 of 2) sorted by relevance
191 TEST(isRowMajorBatchMatmul, Simple) { in TEST() argument204 TEST(isRowMajorBatchMatmul, BindingShifted) { in TEST() argument217 TEST(isRowMajorBatchMatmul, BindingSwapped) { in TEST() argument230 TEST(isRowMajorBatchMatmul, FirstInputSwapped) { in TEST() argument
72 bool mlir::isRowMajorBatchMatmul(ArrayAttr indexingMaps) { in isRowMajorBatchMatmul() function in mlir