Searched defs:collate_fn (Results 1 – 3 of 3) 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 | 237 collate_fn, argument
|
| /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
|