SequenceNumber.h 333 B

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