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