Searched defs:dataset_path (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/pytorch/benchmarks/sparse/dlmc/ |
| H A D | utils.py | 72 def load_spmv_dataset(dataset_path, hidden_size, sparsity, device, n_limit=math.inf): argument
111 dataset_path, hidden_size, sparsity, spmm_type, device, n_limit=math.inf argument
157 dataset_path, argument
|
| H A D | matmul_bench.py | 126 dataset_path = args.path variable
128 dataset_path = os.path.join(dataset_path, dataset_name) variable
|
| /aosp_15_r20/external/executorch/examples/mediatek/model_export_scripts/ |
| H A D | inception_v3.py | 29 def get_dataset(dataset_path, data_size): argument
|
| H A D | inception_v4.py | 29 def get_dataset(dataset_path, data_size): argument
|
| H A D | resnet50.py | 29 def get_dataset(dataset_path, data_size): argument
|
| H A D | mobilenet_v2.py | 29 def get_dataset(dataset_path, data_size): argument
|
| H A D | mobilenet_v3.py | 29 def get_dataset(dataset_path, data_size): argument
|
| H A D | resnet18.py | 29 def get_dataset(dataset_path, data_size): argument
|
| /aosp_15_r20/external/executorch/examples/qualcomm/ |
| H A D | utils.py | 375 dataset_path, data_size, image_shape, crop_size=None, shuffle=True argument
|