ConditionalFeedForward (definition) in projects: aosp_15_r20

Searched defs:ConditionalFeedForward (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/benchmarks/gpt_fast/
H A D	mixtral_moe_model.py	`223 class ConditionalFeedForward(nn.Module): class`
/aosp_15_r20/external/executorch/examples/models/llama/
H A D	llama_transformer.py	`376 class ConditionalFeedForward(nn.Module): class`