#pragma once #include #include // A simple thread local enumeration, used to link forward and backward pass // ops and is used by autograd and observers framework namespace at::sequence_number { TORCH_API uint64_t peek(); TORCH_API uint64_t get_and_increment(); } // namespace at::sequence_number