Searched defs:DropoutState (Results 1 – 1 of 1) sorted by relevance
2287 struct DropoutState { struct2291 at::Tensor buffer;2292 std::optional<cuda::CUDAEvent> event;2293 std::mutex mutex;2297 cuda::CaptureId_t capture_id_last_lock = 0;2298 cuda::CaptureId_t capture_id_last_unlock = 0;2307 void lock() { in lock()2329 void unlock() { in unlock()