Home
last modified time | relevance | path

Searched defs:FeedForward (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/benchmarks/gpt_fast/
Dmodel.py244 class FeedForward(nn.Module): class
/external/executorch/examples/models/llama/
Dllama_transformer.py363 class FeedForward(nn.Module): class
/external/pytorch/torch/testing/_internal/distributed/_tensor/
Dcommon_dtensor.py142 class FeedForward(nn.Module): class