Home
last modified time | relevance | path

Searched refs:torchvision_models (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_fx.py67 from torchvision import models as torchvision_models unknown
1968 rn18 = torchvision_models.resnet18()
1975 rn18 = torchvision_models.resnet18()
2196 rn18 = torchvision_models.resnet18()
4586 model = torchvision_models.get_model(name, **kwargs)
4612 for k in torchvision_models.list_models(module=torchvision_models):
4621 for k in torchvision_models.list_models(module=torchvision_models.segmentation):
4630 for k in torchvision_models.list_models(module=torchvision_models.detection):
4639 for k in torchvision_models.list_models(module=torchvision_models.video):
H A Dtest_flop_counter.py22 from torchvision import models as torchvision_models unknown
250 resnet18 = torchvision_models.resnet18()
/aosp_15_r20/external/pytorch/benchmarks/functional_autograd_benchmark/
H A Dvision_models.py3 import torchvision_models as models
H A DREADME.md51 - `torchaudio_models.py` and `torchvision_models.py` contains code extracted from torchaudio and t…
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dgen_test_model.py41 from torchvision_models import (
H A DREADME.md41 - torchvision_models (https://pytorch.org/vision/stable/models.html)