 |
Tensor Comprehensions
|
Go to the documentation of this file.
20 static constexpr
auto TC_BATCH_MATMUL_NAME =
"batch_matmul";
22 static constexpr
auto TC_BATCH_MATMUL = R
"TC(
def batch_matmul(float(B, N, M) X, float(B, M, K) Y) -> (Z) {
Z(b, n, k) +=! X(b, n, mm) * Y(b, mm, k)
}
)TC";