core.transformer.identity_op

Module Contents

Classes

IdentityOp	This is a placeholder for IdentityOp(x) -> x
IdentityFuncOp	This is a placeholder for IdentityFuncOp(…)(x) -> IdentityOp(x) -> x. Such a func is handy for ops like bias_dropout_fusion which themselves return a function at runtime based on passed arguments

API

class core.transformer.identity_op.IdentityOp(*args, **kwargs)

Bases: torch.nn.Module

This is a placeholder for IdentityOp(x) -> x

Initialization

forward(x, *args, **kwargs)

class core.transformer.identity_op.IdentityFuncOp(*args, **kwargs)

Bases: core.transformer.identity_op.IdentityOp

This is a placeholder for IdentityFuncOp(…)(x) -> IdentityOp(x) -> x. Such a func is handy for ops like bias_dropout_fusion which themselves return a function at runtime based on passed arguments

Initialization

forward(*args, **kwargs)