adding public visibility

core/src/ops/quant.rs

@@ -69,8 +69,8 @@ fn info_quantize_linear_i8(q: &QuantizeLinearI8) -> TractResult<Vec<String>> {
 
 #[derive(Clone, Debug, new)]
 pub struct DequantizeLinearF32 {
-    scale: f32,
-    zero_point: i32,
+    pub scale: f32,
+    pub zero_point: i32,
 }
 
 impl DequantizeLinearF32 {

hir/src/ops/array/flatten.rs

@@ -3,7 +3,7 @@ use crate::internal::*;
 
 #[derive(Debug, Clone, new, Default, Hash)]
 pub struct Flatten {
-    axis: i64,
+    pub axis: i64,
 }
 
 impl Flatten {

hir/src/ops/nn/layer_max.rs

@@ -79,8 +79,8 @@ impl Expansion for LayerHardmax {
 
 #[derive(Debug, Clone, new, Default, Hash)]
 pub struct LayerLogSoftmax {
-    axis: isize,
-    coerce_to_2d: bool,
+    pub axis: isize,
+    pub coerce_to_2d: bool,
 }
 
 

hir/src/ops/nn/mod.rs

@@ -1,7 +1,7 @@
-mod global_pools;
-mod layer_max;
-mod reduce;
-mod softmax;
+pub mod global_pools;
+pub mod layer_max;
+pub mod reduce;
+pub mod softmax;
 
 pub use global_pools::*;
 pub use layer_max::*;

hir/src/ops/nn/reduce.rs

@@ -109,9 +109,9 @@ impl Reducer {
 
 #[derive(Clone, Debug, new, Hash)]
 pub struct Reduce {
-    axes: Option<Vec<i64>>,
-    keep_dims: bool,
-    reducer: Reducer,
+    pub axes: Option<Vec<i64>>,
+    pub keep_dims: bool,
+    pub reducer: Reducer,
 }
 
 

onnx/src/ops/cumsum.rs

@@ -18,9 +18,9 @@ fn cumsum(
 }
 
 #[derive(Debug, Clone, Hash)]
-struct CumSum {
-    reverse: bool,
-    exclusive: bool,
+pub struct CumSum {
+    pub reverse: bool,
+    pub exclusive: bool,
 }
 
 impl Expansion for CumSum {

onnx/src/ops/mod.rs

@@ -4,7 +4,7 @@ use tract_hir::internal::*;
 
 mod array;
 mod cast;
-mod cumsum;
+pub mod cumsum;
 mod d2s;
 mod einsum;
 mod fft;

onnx/src/ops/nn/reduce.rs

@@ -27,10 +27,10 @@ pub(crate) fn reduce(
 
 #[derive(Debug, Clone, Hash)]
 struct ReduceSum13 {
-    have_axis_input: bool,
-    keep_dims: bool,
-    noop_with_empty_axes: bool,
-    reducer: tract_hir::ops::nn::Reducer,
+    pub have_axis_input: bool,
+    pub keep_dims: bool,
+    pub noop_with_empty_axes: bool,
+    pub reducer: tract_hir::ops::nn::Reducer,
 }