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