Home
last modified time | relevance | path

Searched +path:pytorch +path:test +path:nn (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_parametrization.py
H A Dtest_load_state_dict.py
H A Dtest_packed_sequence.py
H A Dtest_module_hooks.py
H A Dtest_convolution.py
H A Dtest_pruning.py
H A Dtest_embedding.py
H A Dtest_dropout.py
H A Dtest_multihead_attention.py
H A Dtest_init.py
H A Dtest_pooling.py
H A Dtest_lazy_modules.py
/aosp_15_r20/external/pytorch/test/distributed/nn/jit/
H A D__init__.py
H A Dtest_instantiator.py