Class PoolingWithMaskingOp

Inheritance Relationships

Base Type

Class Documentation

class PoolingWithMaskingOp : public marian::UnaryNodeOp

Public Functions

PoolingWithMaskingOp(Expr x, Expr mask, int width, bool isEven = false)
NodeOps forwardOps()
NodeOps backwardOps()
const std::string type()

Protected Attributes

Expr mask_
int width_
bool isEven_

Friends

friend marian::PoolingWithMaskingOp::SerializationHelpers