fcrelu.h

Go to the documentation of this file.

#pragma once

namespace tc {

constexpr static auto TC_FCRELU_NAME = "func_fcrelu";

constexpr static auto TC_FCRELU = R"TC(
  def func_fcrelu(float(B,M) I, float(N,M) W1, float(N) B1) -> (O1) {
    O1(b, n) +=! I(b, m) * W1(n, m)
    O1(b, n) = O1(b, n) + B1(n)
    O1(b, n) = fmax(O1(b, n), 0)
  }
)TC";

} // namespace tc