Class AffineWithReluNodeOp

Inheritance Relationships

Base Type

Class Documentation

class AffineWithReluNodeOp : public marian::NaryNodeOp

Public Functions

AffineWithReluNodeOp(Expr a, Expr b, Expr bias, bool transA, bool transB, float scalar)
Shape newShape(Expr a, Expr b, bool transA, bool transB)
NodeOps forwardOps()
NodeOps backwardOps()
const std::string type()
virtual size_t hash()
virtual bool equal(Expr node)