Searched defs:_streams (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/csrc/cuda/ |
| 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
|
| /external/pytorch/torch/nn/parallel/ |
| D | _functions.py | 120 _streams: Optional[List[Optional[torch.Stream]]] = None variable
|