Home
last modified time | relevance | path

Searched defs:CUDAEvent (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCUDAEvent.h28 struct TORCH_CUDA_CPP_API CUDAEvent { struct
32 CUDAEvent(unsigned int flags) noexcept : flags_{flags} {} in CUDAEvent() argument
34 CUDAEvent( in CUDAEvent() function
44 ~CUDAEvent() { in ~CUDAEvent() argument
58 CUDAEvent& operator=(const CUDAEvent&) = delete; argument
60 CUDAEvent(CUDAEvent&& other) noexcept { moveHelper(std::move(other)); } in CUDAEvent() function
71 friend bool operator<(const CUDAEvent& left, const CUDAEvent& right) { argument
75 std::optional<at::Device> device() const { in device()
83 bool isCreated() const { return is_created_; } in isCreated()
84 DeviceIndex device_index() const {return device_index_;} in device_index()
[all …]
/aosp_15_r20/external/pytorch/torch/csrc/jit/cuda/
H A Dcuda.h8 class CUDAEvent; variable
61 friend class CUDAEvent; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_event.h24 using CUDAEvent = gpu::GpuEvent; variable