(a, b, matrix_bias, transa, transb)
| 43 | ) |
| 44 | |
| 45 | def get_numpy(a, b, matrix_bias, transa, transb): |
| 46 | if transa: |
| 47 | a = a.transpose() |
| 48 | if transb: |
| 49 | b = b.transpose() |
| 50 | return np.dot(a, b) + matrix_bias |
| 51 | |
| 52 | def compiling(f, name="test_matmul_add", ext=".so"): |
| 53 | path = name + ext |
no test coverage detected
searching dependent graphs…