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

test: zero out first input for custom-ops tests

Browse Source
This commit is contained in:
Carson M.
2024-12-21 01:01:15 -06:00
parent cf447cf566
commit 4d16d600d5
2 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
examples/custom-ops/examples/custom-ops.rs
View File

@@ -79,7 +79,13 @@ fn main() -> ort::Result<()> {
.commit_from_file("tests/data/custom_op_test.onnx")?;
let allocator = session.allocator();
let value1 = Tensor::<f32>::new(allocator, [3, 5])?;
let mut value1 = Tensor::<f32>::new(allocator, [3, 5])?;
{
let (_, data) = value1.extract_raw_tensor_mut();
for datum in data {
*datum = 0.;
}
}
let mut value2 = Tensor::<f32>::new(allocator, [3, 5])?;
{
let (_, data) = value2.extract_raw_tensor_mut();

8
src/operator/tests.rs
View File

@@ -82,7 +82,13 @@ fn test_custom_ops() -> crate::Result<()> {
.commit_from_file("tests/data/custom_op_test.onnx")?;
let allocator = session.allocator();
let value1 = Tensor::<f32>::new(allocator, [3, 5])?;
let mut value1 = Tensor::<f32>::new(allocator, [3, 5])?;
{
let (_, data) = value1.extract_raw_tensor_mut();
for datum in data {
*datum = 0.;
}
}
let mut value2 = Tensor::<f32>::new(allocator, [3, 5])?;
{
let (_, data) = value2.extract_raw_tensor_mut();