MCPcopy Create free account
hub / github.com/OpenDriveLab/ReSim / layernorm.py

File layernorm.py

SwissArmyTransformer/sat/ops/layernorm.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1try:
2 from apex.normalization.fused_layer_norm import FusedLayerNorm
3 class LayerNorm(FusedLayerNorm):
4 def __init__(self, *args, pb_relax=False, **kwargs):

Callers

nothing calls this directly

Calls 1

print_rank0Function · 0.90

Tested by

no test coverage detected