2fcrelu.h

Go to the documentation of this file.

#pragma once

namespace tc {
static constexpr auto TC_2FCRELU_NAME = "func_2fcrelu";

static constexpr auto TC_2FCRELU = R"TC(
  def func_2fcrelu(float(B,M) I, float(N,M) W1, float(N) B1, float(O,N) W2, float(O) B2) -> (O1, O2) {
    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)
    O2(b, o) +=! O1(b, n) * W2(o, n)
    O2(b, o) = O2(b, o) + B2(o)
    O2(b, o) = fmax(O2(b, o), 0)
  }
)TC";
} // namespace tc