model.py File Reference#
|
SDK qb Runtime Library v1.0
MCS001-
|
Go to the source code of this file.
Classes | |
| class | qbruntime.model.Model |
| Represents an AI model loaded from an MXQ file. More... | |
Functions | |
| bool | qbruntime.model._is_valid_shape (_Shape input_shape, _Shape shape) |
| Optional[Tuple[bool, bool]] | qbruntime.model._find_memory_format (List[np.ndarray] inputs, List[_Shape] shapes) |
| Tuple[int, Tuple[bool, bool]] | qbruntime.model._find_matching_variant_idx_and_memory_format (model, List[np.ndarray] inputs) |
| List[np.ndarray] | qbruntime.model._build_outputs (List[_Shape] shapes, bool is_hwc, np.dtype dtype) |
| None | qbruntime.model._check_output_shapes (List[np.ndarray] outputs, List[_Shape] shapes, bool is_hwc, np.dtype dtype) |
| Model | qbruntime.model.load (str path, Optional[ModelConfig] model_config=None) |
| Single-step inference API. | |
Variables | |
| qbruntime.model._Shape = Tuple[int, ...] | |
Variable Documentation
◆ _Shape
Generated by