Class FbgemmPacked16AffineNodeOp

• Defined in File expanded_gemm.h

Inheritance Relationships

Base Type

• public marian::NaryNodeOp (Struct NaryNodeOp)

Class Documentation

class FbgemmPacked16AffineNodeOp : public marian::NaryNodeOp

Public Functions

FbgemmPacked16AffineNodeOp(const std::vector<Expr> &nodes, Shape bShape, bool transA, bool transB, float)

Shape newShape(Expr a, Shape bShape, bool transA, bool transB)

NodeOps forwardOps()

NodeOps backwardOps()

const std::string type()