eliott/ort
1
0
Fork 0
mirror of https://github.com/pykeio/ort synced 2026-04-25 16:34:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix: well that's not right

Browse Source
This commit is contained in:
Carson M.
2026-03-11 14:26:27 -05:00
parent e336d6bf05
commit 14e9d29f49
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/value/impl_tensor/element.rs
View File

@@ -63,7 +63,7 @@ impl TensorElementType {
TensorElementType::Int64 | TensorElementType::Uint64 => container_capacity * 8, TensorElementType::Int64 | TensorElementType::Uint64 => container_capacity * 8,
TensorElementType::String => 0, // unsure what to do about this... TensorElementType::String => 0, // unsure what to do about this...
TensorElementType::Float8E4M3FN | TensorElementType::Float8E4M3FNUZ | TensorElementType::Float8E5M2 | TensorElementType::Float8E5M2FNUZ => { TensorElementType::Float8E4M3FN | TensorElementType::Float8E4M3FNUZ | TensorElementType::Float8E5M2 | TensorElementType::Float8E5M2FNUZ => {
container_capacity * 4 container_capacity
} }
TensorElementType::Float16 | TensorElementType::Bfloat16 => container_capacity * 2, TensorElementType::Float16 | TensorElementType::Bfloat16 => container_capacity * 2,
TensorElementType::Float32 => container_capacity * 4, TensorElementType::Float32 => container_capacity * 4,