(dtype)
| 123 | |
| 124 | @contextlib.contextmanager |
| 125 | def precision(dtype): |
| 126 | if isinstance(dtype, str): |
| 127 | dtype = str_dtype_to_trt(dtype) |
| 128 | prev_dtype = switch_net_dtype(dtype) |
| 129 | yield |
| 130 | switch_net_dtype(prev_dtype) |
| 131 | |
| 132 | |
| 133 | def serialize_engine(engine, path): |