model.py File Reference

model.py File Reference#

SDK qb Runtime Library: model.py File Reference
SDK qb Runtime Library v1.0
MCS001-
model.py File Reference

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

qbruntime.model._Shape = Tuple[int, ...]
protected

Definition at line 15 of file model.py.