patch conv

zkonduit · Feb 8, 2025 · 3cd9d2a · 3cd9d2a
1 parent 8f0c460
commit 3cd9d2a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/benches/accum_conv.rs b/benches/accum_conv.rs
@@ -73,6 +73,8 @@ impl Circuit<Fr> for MyCircuit {
                             padding: vec![(0, 0)],
                             stride: vec![1; 2],
                             group: 1,
+                            data_format: DataFormat::NCHW,
+                            kernel_format: KernelFormat::OIHW,
                         }),
                     )
                     .unwrap();

diff --git a/examples/conv2d_mnist/main.rs b/examples/conv2d_mnist/main.rs
@@ -31,6 +31,7 @@ use instant::Instant;
 use mnist::*;
 use rand::rngs::OsRng;
 use std::marker::PhantomData;
+use tract_onnx::tract_core::ndarray::Data;
 
 mod params;
 
@@ -208,6 +209,8 @@ where
                         padding: vec![(PADDING, PADDING); 2],
                         stride: vec![STRIDE; 2],
                         group: 1,
+                        data_format: DataFormat::NCHW,
+                        kernel_format: KernelFormat::OIHW,
                     };
                     let x = config
                         .layer_config