| 76 | } |
| 77 | |
| 78 | void quantize_bf16(program& prog, const std::vector<std::string>& ins_names) |
| 79 | { |
| 80 | run_passes(prog, |
| 81 | {normalize_ops{}, |
| 82 | optimize_module{{"quantizelinear", "dequantizelinear"}}, |
| 83 | truncate_float_pass{ins_names, shape::bf16_type}, |
| 84 | optimize_module{{"quantizelinear", "dequantizelinear"}}}, |
| 85 | quant_tracer()); |
| 86 | } |
| 87 | |
| 88 | static void quantize_8bits(program& prog, |
| 89 | const target& t, |