uor-prism-tensor 0.3.1

Prism standard-library tensor-compute sub-crate (wiki ADR-031): declares TensorAxis and ActivationAxis under the `axis!` SDK macro with CPU integer and fixed-point reference impls preserving bit-determinism per (H, B, A) selection.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147

//! Higher-rank tensor `ConstrainedTypeShape` carriers
//! ([`Tensor3Shape`], [`Tensor4Shape`]).
//!
//! Per [Wiki ADR-031][09]'s `Tensor<Element, Shape>` shape commitment,
//! the tensor sub-crate declares typed shape carriers per common rank:
//!
//! - rank-1 → [`VectorShape`][crate::tensor::VectorShape]
//! - rank-2 → [`MatrixShape`][crate::tensor::MatrixShape]
//! - rank-3 → [`Tensor3Shape`] *(this module)*
//! - rank-4 → [`Tensor4Shape`] *(this module)*
//!
//! Higher ranks compose through `partition_product!` per ADR-033/044;
//! the carriers in this module cover the common GGUF / ONNX tensor
//! ranks (embeddings, attention heads, batched attention) directly.
//!
//! Per [ADR-017][09]'s closure rule each carrier shares the generic
//! `https://uor.foundation/type/ConstrainedType` IRI and content-
//! addresses through `(SITE_COUNT, CONSTRAINTS)`; the Rust-type
//! distinction is the application-level ergonomics surface for
//! variable-rank tensor shapes.
//!
//! [09]: https://github.com/UOR-Foundation/UOR-Framework/wiki/09-Architecture-Decisions

#![allow(missing_docs)]

use uor_foundation::enforcement::GroundedShape;
use uor_foundation::pipeline::{ConstrainedTypeShape, ConstraintRef, IntoBindingValue, TermValue};

/// Parametric `ConstrainedTypeShape` for a row-major rank-3 tensor of
/// shape `D0 × D1 × D2` carrying `ELEM_BYTES`-byte elements.
///
/// Per ADR-031's `Tensor<Element, Shape>` shape commitment for rank-3.
/// Common GGUF / ONNX usage: per-head attention key / value tensors
/// (`batch × heads × dim`), 3D image volumes, sequence-of-tokens
/// embeddings.
#[derive(Debug, Clone, Copy)]
pub struct Tensor3Shape<const D0: usize, const D1: usize, const D2: usize, const ELEM_BYTES: usize>;

impl<const D0: usize, const D1: usize, const D2: usize, const ELEM_BYTES: usize> Default
    for Tensor3Shape<D0, D1, D2, ELEM_BYTES>
{
    fn default() -> Self {
        Self
    }
}

impl<const D0: usize, const D1: usize, const D2: usize, const ELEM_BYTES: usize>
    ConstrainedTypeShape for Tensor3Shape<D0, D1, D2, ELEM_BYTES>
{
    const IRI: &'static str = "https://uor.foundation/type/ConstrainedType";
    const SITE_COUNT: usize = D0 * D1 * D2 * ELEM_BYTES;
    const CONSTRAINTS: &'static [ConstraintRef] = &[];
    #[allow(clippy::cast_possible_truncation)]
    const CYCLE_SIZE: u64 = 256u64.saturating_pow((D0 * D1 * D2 * ELEM_BYTES) as u32);
}

impl<const D0: usize, const D1: usize, const D2: usize, const ELEM_BYTES: usize>
    uor_foundation::pipeline::__sdk_seal::Sealed for Tensor3Shape<D0, D1, D2, ELEM_BYTES>
{
}
impl<const D0: usize, const D1: usize, const D2: usize, const ELEM_BYTES: usize> GroundedShape
    for Tensor3Shape<D0, D1, D2, ELEM_BYTES>
{
}
impl<'a, const D0: usize, const D1: usize, const D2: usize, const ELEM_BYTES: usize>
    IntoBindingValue<'a> for Tensor3Shape<D0, D1, D2, ELEM_BYTES>
{
    fn as_binding_value<const INLINE_BYTES: usize>(&self) -> TermValue<'a, INLINE_BYTES> {
        TermValue::empty()
    }
}

/// Parametric `ConstrainedTypeShape` for a row-major rank-4 tensor of
/// shape `D0 × D1 × D2 × D3` carrying `ELEM_BYTES`-byte elements.
///
/// Per ADR-031's `Tensor<Element, Shape>` shape commitment for rank-4.
/// Common GGUF / ONNX usage: batched multi-head attention
/// (`batch × heads × seq × dim`), 4D conv weight tensors
/// (`out_channels × in_channels × kernel_h × kernel_w`).
#[derive(Debug, Clone, Copy)]
pub struct Tensor4Shape<
    const D0: usize,
    const D1: usize,
    const D2: usize,
    const D3: usize,
    const ELEM_BYTES: usize,
>;

impl<
        const D0: usize,
        const D1: usize,
        const D2: usize,
        const D3: usize,
        const ELEM_BYTES: usize,
    > Default for Tensor4Shape<D0, D1, D2, D3, ELEM_BYTES>
{
    fn default() -> Self {
        Self
    }
}

impl<
        const D0: usize,
        const D1: usize,
        const D2: usize,
        const D3: usize,
        const ELEM_BYTES: usize,
    > ConstrainedTypeShape for Tensor4Shape<D0, D1, D2, D3, ELEM_BYTES>
{
    const IRI: &'static str = "https://uor.foundation/type/ConstrainedType";
    const SITE_COUNT: usize = D0 * D1 * D2 * D3 * ELEM_BYTES;
    const CONSTRAINTS: &'static [ConstraintRef] = &[];
    #[allow(clippy::cast_possible_truncation)]
    const CYCLE_SIZE: u64 = 256u64.saturating_pow((D0 * D1 * D2 * D3 * ELEM_BYTES) as u32);
}

impl<
        const D0: usize,
        const D1: usize,
        const D2: usize,
        const D3: usize,
        const ELEM_BYTES: usize,
    > uor_foundation::pipeline::__sdk_seal::Sealed for Tensor4Shape<D0, D1, D2, D3, ELEM_BYTES>
{
}
impl<
        const D0: usize,
        const D1: usize,
        const D2: usize,
        const D3: usize,
        const ELEM_BYTES: usize,
    > GroundedShape for Tensor4Shape<D0, D1, D2, D3, ELEM_BYTES>
{
}
impl<
        'a,
        const D0: usize,
        const D1: usize,
        const D2: usize,
        const D3: usize,
        const ELEM_BYTES: usize,
    > IntoBindingValue<'a> for Tensor4Shape<D0, D1, D2, D3, ELEM_BYTES>
{
    fn as_binding_value<const INLINE_BYTES: usize>(&self) -> TermValue<'a, INLINE_BYTES> {
        TermValue::empty()
    }
}