1 #pragma once 2 3 #include <c10/macros/Export.h> 4 #include <cstdint> 5 6 // A simple thread local enumeration, used to link forward and backward pass 7 // ops and is used by autograd and observers framework 8 namespace at::sequence_number { 9 10 TORCH_API uint64_t peek(); 11 TORCH_API uint64_t get_and_increment(); 12 13 } // namespace at::sequence_number 14