Home
last modified time | relevance | path

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

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