| 12345678910111213141516 |
- #pragma once
- #include <c10/macros/Export.h>
- #include <cstdint>
- namespace at {
- // A simple thread local enumeration, used to link forward and backward pass
- // ops and is used by autograd and observers framework
- namespace sequence_number {
- TORCH_API uint64_t peek();
- TORCH_API uint64_t get_and_increment();
- } // namespace sequence_number
- } // namespace at
|
