diff options
Diffstat (limited to 'aten/src/ATen/Generator.h')
-rw-r--r-- | aten/src/ATen/Generator.h | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/aten/src/ATen/Generator.h b/aten/src/ATen/Generator.h new file mode 100644 index 0000000000..a99da0e406 --- /dev/null +++ b/aten/src/ATen/Generator.h @@ -0,0 +1,18 @@ +#pragma once + +namespace at { + +struct Generator { + Generator() {}; + Generator(const Generator& other) = delete; + Generator(Generator&& other) = delete; + virtual ~Generator() {}; + + virtual Generator& copy(const Generator& other) = 0; + virtual Generator& free() = 0; + + virtual unsigned long seed() = 0; + virtual Generator& manualSeed(unsigned long seed) = 0; +}; + +} // namespace at |