Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dtransformer.cpp32 int64_t dim_feedforward = 16; in get_a_test_layer() local
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_repros.py850 def __init__(self, d_model, dim_feedforward, activation, dropout) -> None: argument
869 dim_feedforward=2048, argument