| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | dataloader.cpp | 167 auto data_loader = torch::data::make_data_loader( in TEST() local
771 auto data_loader = torch::data::make_data_loader( in TEST() local
835 auto data_loader = torch::data::make_data_loader( in TEST() local
1046 auto data_loader = torch::data::make_data_loader( in TEST() local
1069 auto data_loader = torch::data::make_data_loader(DummyDataset(), 32); in TEST() local
1077 auto data_loader = torch::data::make_data_loader(DummyDataset(), 32); in TEST() local
1086 auto data_loader = torch::data::make_data_loader(DummyDataset(), 32); in TEST() local
1094 auto data_loader = in TEST() local
1111 auto data_loader = in TEST() local
1128 auto data_loader = in TEST() local
[all …]
|
| H A D | integration.cpp | 120 auto data_loader = in test_mnist() local
|
| /aosp_15_r20/external/pytorch/test/cpp/aoti_inference/ |
| H A D | test.cpp | 24 torch::jit::script::Module data_loader = torch::jit::load(data_path); in test_aoti() local
89 torch::jit::script::Module data_loader = torch::jit::load(data_path); in test_aoti_constants_update() local
178 torch::jit::script::Module data_loader = torch::jit::load(data_path); in test_aoti_double_buffering() local
259 torch::jit::script::Module data_loader = torch::jit::load(data_path); in test_aoti_double_buffering_with_tensor_constants() local
|
| /aosp_15_r20/external/executorch/runtime/executor/test/ |
| H A D | program_test.cpp | 164 BufferDataLoader data_loader(data.get(), data_len); in TEST_F() local
229 BufferDataLoader data_loader(data.get(), data_len); in TEST_F() local
254 BufferDataLoader data_loader(data.get() + 1, data_len); in TEST_F() local
|
| /aosp_15_r20/external/pytorch/test/quantization/core/experimental/ |
| H A D | quantization_util.py | 72 def evaluate(model, criterion, data_loader): argument
132 def training_loop(model, criterion, data_loader): argument
|
| H A D | apot_fx_graph_mode_ptq.py | 13 data_loader, data_loader_test = prepare_data_loaders(data_path) variable
31 def calibrate(model, data_loader): argument
|
| H A D | apot_fx_graph_mode_qat.py | 14 data_loader, data_loader_test = prepare_data_loaders(data_path) variable
|
| /aosp_15_r20/external/executorch/extension/training/module/ |
| H A D | training_module.h | 37 std::move(data_loader), in Module()
|
| /aosp_15_r20/external/executorch/extension/module/test/ |
| H A D | module_test.cpp | 275 auto data_loader = std::make_unique<FileDataLoader>(std::move(loader.get())); in TEST_F() local
306 auto data_loader = in TEST_F() local
|
| /aosp_15_r20/external/executorch/extension/module/ |
| H A D | module.cpp | 52 std::unique_ptr<runtime::DataLoader> data_loader, in Module()
|
| /aosp_15_r20/external/pytorch/test/cpp/jit/ |
| H A D | test_lite_trainer.cpp | 331 auto data_loader = torch::data::make_data_loader<mobile::SequentialSampler>( in TEST() local
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_quantization.py | 193 def train_one_epoch(model, criterion, optimizer, data_loader, device, ntrain_batches): argument
|