Searched defs:drop_last (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/utils/data/_utils/ |
| H A D | fetch.py | 9 def __init__(self, dataset, auto_collation, collate_fn, drop_last): argument
20 def __init__(self, dataset, auto_collation, collate_fn, drop_last): argument
|
| H A D | worker.py | 238 drop_last, argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/data/ |
| H A D | dataloader_options.h | 62 bool drop_last; member
|
| /aosp_15_r20/external/pytorch/torch/utils/data/ |
| H A D | dataloader.py | 75 def create_fetcher(kind, dataset, auto_collation, collate_fn, drop_last): argument
|