Home
last modified time | relevance | path

Searched defs:enable_timing (Results 1 – 9 of 9) sorted by relevance

/external/pytorch/torch/mps/
Devent.py16 def __init__(self, enable_timing=False): argument
/external/pytorch/benchmarks/sparse/
Dutils.py13 def __init__(self, enable_timing): argument
/external/pytorch/torch/csrc/xpu/
DEvent.cpp20 unsigned char enable_timing = 0; in THXPEvent_pynew() local
/external/pytorch/torch/xpu/
Dstreams.py116 def __new__(cls, enable_timing=False): argument
/external/pytorch/torch/cuda/
Dstreams.py164 def __new__(cls, enable_timing=False, blocking=False, interprocess=False): argument
/external/pytorch/torch/csrc/cuda/
DEvent.cpp23 unsigned char enable_timing = 0; in THCPEvent_pynew() local
/external/pytorch/benchmarks/fastrnns/
Dbench.py59 def __init__(self, enable_timing): argument
/external/pytorch/torch/csrc/
DEvent.cpp26 unsigned char enable_timing = 0; in THPEvent_pynew() local
/external/pytorch/torch/csrc/mps/
DModule.cpp161 const bool enable_timing = THPUtils_unpackBool(args); in MPSModule_acquireEvent() local