Searched refs:test_dataloader (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/data_sparsifier/benchmarks/ |
| H A D | evaluate_model_metrics.py | 18 def inference_and_evaluation(dlrm, test_dataloader, device): argument
32 nbatches = len(test_dataloader)
36 for i, testBatch in enumerate(test_dataloader):
82 def evaluate_metrics(test_dataloader, sparse_model_metadata): argument
107 model_metrics = inference_and_evaluation(model, test_dataloader, device)
136 test_dataloader = make_test_data_loader( variable
141 evaluate_metrics(test_dataloader, args.sparse_model_metadata)
|
| /aosp_15_r20/external/pytorch/.ci/docker/ |
| H A D | requirements-ci.txt | 31 #test that import: dynamo/test_replay_record.py test_dataloader.py test_datapipe.py test_serializat…
129 #test_jit.py, test_indexing.py, test_datapipe.py, test_dataloader.py,
168 #test that import: test_profiler.py, test_openmp.py, test_dataloader.py
|
| /aosp_15_r20/external/libopus/dnn/torch/neural-pitch/ |
| H A D | training.py | 92 test_dataloader = torch.utils.data.DataLoader(dataset=test_dataset, batch_size=batch_size, shuffle=… variable
126 with tqdm.tqdm(test_dataloader) as test_epoch:
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse.py | 5315 def test_dataloader(self, device, layout, dtype): member in TestSparseAny
|