Home
last modified time | relevance | path

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

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