File size: 333 Bytes
9c6594c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
#pragma once
#include <c10/macros/Export.h>
#include <cstdint>
// 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
|