1 | #include <torch/csrc/jit/passes/quantization/quantization_type.h> |
---|---|
2 | |
3 | namespace torch { |
4 | namespace jit { |
5 | |
6 | std::ostream& operator<<(std::ostream& os, QuantType t) { |
7 | switch (t) { |
8 | case QuantType::DYNAMIC: |
9 | os << "dynamic"; |
10 | break; |
11 | case QuantType::STATIC: |
12 | os << "static"; |
13 | break; |
14 | default: |
15 | os.setstate(std::ios_base::failbit); |
16 | } |
17 | return os; |
18 | } |
19 | |
20 | } // namespace jit |
21 | } // namespace torch |
22 |