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: getting attribute size in ShapeInferenceContext::attr

Browse Source
This commit is contained in:
Carson M.
2026-03-20 19:42:41 -05:00
parent 477e929495
commit 9557b6b03f
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/operator/mod.rs
View File

@@ -94,7 +94,9 @@ impl ShapeInferenceContext {
})?;
let mut len = 0;
ortsys![unsafe ReadOpAttr(attr, T::attr_type(), ptr::null_mut(), 0, &mut len)?];
// this will always error since the length we pass (0) is too small to hold anything; we don't care about value yet, we
// just want the real length which will end up in the `len` variable
let _ = ortsys![@ort: unsafe ReadOpAttr(attr, T::attr_type(), ptr::null_mut(), 0, &mut len) as Result];
unsafe { T::from_op_attr(attr, len) }
}