types.h File Reference

Basic types of the blace.ai library. More...

#include "library_defines.h"
#include <cstdint>
#include <iosfwd>
#include <optional>
#include <string>
#include <vector>

Go to the source code of this file.

Classes
struct	blace::ml_core::LAB_NORMS
struct	blace::ml_core::BlaceHash
struct	blace::ml_core::ModelInferenceArgs
struct	blace::ml_core::MultisampleInferenceArgs
struct	blace::ml_core::SlicedInferenceArgs
class	blace::ml_core::InferenceArgsCollection
class	blace::ml_core::ModelIdent

Typedefs
typedef std::vector< int64_t >	blace::ml_core::TensorSizes

Enumerations
enum	blace::ml_core::DeviceEnum : int { CPU = 0 , CUDA = 1 , MPS = 2 , DML = 3 }
enum	blace::ml_core::ValueRangeEnum : int {   ZERO_TO_ONE = 0 , MINUS_ONE_TO_ONE = 1 , IMAGENET = 2 , UNKNOWN_VALUE_RANGE = 3 ,   ZERO_TO_255 = 4 , MINUS_0_5_TO_0_5 = 5 , ZERO_TO_32768 = 6 }
enum	blace::ml_core::DataTypeEnum : int {   INT_32 = 0 , FLOAT_32 = 1 , BLACE_BYTE = 2 , BLACE_BOOL = 3 ,   FLOAT_32_16 = 4 , FLOAT_16 = 5 , SHORT = 6 , INT_64 = 7 ,   FLOAT_64 = 8 }
enum	blace::ml_core::OrderEnum : int {   BTCHW = 0 , BCHW = 1 , CHW = 2 , HWC = 3 ,   BHWC = 4 , HW = 5 , W = 6 , WC = 7 ,   C = 8 , BC = 9 , BWCH = 10 , BHW = 11 ,   BCH = 12 , CH = 13 , TBCHW = 14 , BCWH = 15 ,   BWHC = 16 , NO_DIMS = 17 , UNKNOWN_ORDER = 18 , BOUNDING_BOX_WITH_DIMS = 19 ,   THWC = 20 , TCHW = 21 , H = 22 , BAUDIO_MONO_44100HZ = 23 }
enum	blace::ml_core::ColorFormatEnum : int {   RGB = 0 , R = 1 , A = 2 , ARGB = 3 ,   ARBITRARY_CHANNELS = 4 , BGRA = 5 , BGR = 6 , LAB = 7 ,   AB = 8 , L = 9 , XYZ = 10 , YCBCR = 11 ,   Y = 12 , BRG = 13 , UV = 14 }
enum	blace::ml_core::ModelDataType { tFloat = 0 , tHalf }
enum	blace::ml_core::FORWARD_MODE { SIMPLE = 0 , MULTIPASS , SLICED }
enum	blace::ml_core::Interpolation {   NEAREST = 0 , LINEAR , BILINEAR , BICUBIC ,   TRILINEAR , AREA , PIL_BICUBIC }
enum	blace::ml_core::PADDING_MODE { REPLICATION = 0 , REFLECTION , CONSTANT }
enum	blace::ml_core::DIRECTION { LEFT , TOP , RIGHT , BOTTOM }
enum	blace::ml_core::Backend {   TORCHSCRIPT_CUDA_FP32 , TORCHSCRIPT_CUDA_FP16 , TORCHSCRIPT_MPS_FP32 , ONNX_CUDA_FP32 ,   ONNX_CUDA_FP16 , ONNX_CPU_FP32 , TORCHSCRIPT_MPS_FP16 , TORCHSCRIPT_CPU_FP32 ,   ONNX_DML_FP32 , ONNX_DML_FP16 }
enum	blace::ml_core::Framework { TORCHSCRIPT , ONNXRUNTIME }
enum class	blace::ml_core::ReturnCode { OK = 0 , Error , UserCancelled , MissingData }

Variables
constexpr int	blace::ml_core::BYTE_MAX = 255
constexpr float	blace::ml_core::SHORT_MAX = 32768.
constexpr float	blace::ml_core::BYTE_MAX_FLOAT = 255.

Detailed Description

Basic types of the blace.ai library.

Typedef Documentation

TensorSizes

typedef std::vector<int64_t> blace::ml_core::TensorSizes

Tensor sizes.

Enumeration Type Documentation

Backend

enum blace::ml_core::Backend

Backends for model inference. A backend is composed of a provider, a device and a precision.

ColorFormatEnum

enum blace::ml_core::ColorFormatEnum : int

Color format of a tensor.

DataTypeEnum

enum blace::ml_core::DataTypeEnum : int

Data types of a tensor.

DeviceEnum

enum blace::ml_core::DeviceEnum : int

Device enumeration. Use CPU for cpu inference, CUDA for cuda accelerated inference on Windows and Linux systems with NVidia GPU and MPS for metal acceleration on MacOS.

DIRECTION

enum blace::ml_core::DIRECTION

Not used.

FORWARD_MODE

enum blace::ml_core::FORWARD_MODE

Model inference modes.

Framework

enum blace::ml_core::Framework

Available Frameworks for model inference.

Interpolation

enum blace::ml_core::Interpolation

Tensor interpolation modes.

ModelDataType

enum blace::ml_core::ModelDataType

Model data type.

OrderEnum

enum blace::ml_core::OrderEnum : int

Order of a tensor.

PADDING_MODE

enum blace::ml_core::PADDING_MODE

Tensor padding modes.

ReturnCode

enum class blace::ml_core::ReturnCode

strong

Return codes.

ValueRangeEnum

enum blace::ml_core::ValueRangeEnum : int

Enum with different options for value ranges inside a tensor.

Variable Documentation

BYTE_MAX

constexpr int blace::ml_core::BYTE_MAX = 255

constexpr

Maximum value of a byte.

BYTE_MAX_FLOAT

constexpr float blace::ml_core::BYTE_MAX_FLOAT = 255.

constexpr

Maximum value of a byte as float.

SHORT_MAX

constexpr float blace::ml_core::SHORT_MAX = 32768.

constexpr

Maximum value of a short.