batch_matmul.h

Go to the documentation of this file.

#pragma once

namespace tc {

static constexpr auto TC_BATCH_MATMUL_NAME = "batch_matmul";

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";

} // namespace tc