Searched defs:_streams (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/cuda/ |
| H A D | python_nccl.cpp | 132 PyObject *_inputs = nullptr, *_output = nullptr, *_streams = nullptr, in THCPModule_nccl_reduce() local
165 PyObject *_inputs = nullptr, *_outputs = nullptr, *_streams = nullptr, in THCPModule_nccl_all_reduce() local
198 PyObject *_inputs = nullptr, *_streams = nullptr, *_comms = nullptr; in THCPModule_nccl_broadcast() local
229 PyObject *_inputs = nullptr, *_outputs = nullptr, *_streams = nullptr, in THCPModule_nccl_all_gather() local
261 PyObject *_inputs = nullptr, *_outputs = nullptr, *_streams = nullptr, in THCPModule_nccl_reduce_scatter() local
|
| /aosp_15_r20/external/pytorch/torch/nn/parallel/ |
| H A D | _functions.py | 120 _streams: Optional[List[Optional[torch.Stream]]] = None variable
|
| /aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/Llama-3.2-1B-Instruct/ |
| H A D | tokenizer.json | 78384 "_streams": 75972, number
|
| /aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/Llama-3.2-3B-Instruct/ |
| H A D | tokenizer.json | 78384 "_streams": 75972, number
|
| /aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/llama3-8B-instruct/ |
| H A D | tokenizer.json | 78325 "_streams": 75972, number
|