Tag

PyTorch internal tags for compiler/dispatcher.

PyTorch

API: torch.Tag
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: nnx.filterlib.WithTag.tag
Strategy: Direct Mapping
Official Docs ↗