1 | #pragma once |
---|---|
2 | #include <ostream> |
3 | |
4 | namespace torch { |
5 | namespace jit { |
6 | |
7 | // Quantization type (dynamic quantization, static quantization). |
8 | // Should match the Python enum in quantize_jit.py |
9 | enum QuantType : uint8_t { DYNAMIC = 0, STATIC }; |
10 | |
11 | std::ostream& operator<<(std::ostream& os, QuantType t); |
12 | |
13 | } // namespace jit |
14 | } // namespace torch |
15 |