From b9e11738e75634bb3dede1aa7ef345f0568fccfa Mon Sep 17 00:00:00 2001 From: Jan Marais Date: Tue, 8 Oct 2024 08:53:27 +0200 Subject: [PATCH] expose tensorops and error to root --- crates/kornia-core-ops/src/lib.rs | 3 +++ crates/kornia-core-ops/src/ops.rs | 6 +++--- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/crates/kornia-core-ops/src/lib.rs b/crates/kornia-core-ops/src/lib.rs index 4494ce6f..a3a8c878 100644 --- a/crates/kornia-core-ops/src/lib.rs +++ b/crates/kornia-core-ops/src/lib.rs @@ -6,3 +6,6 @@ pub mod error; /// module containing ops implementations. pub mod ops; + +pub use crate::error::TensorOpsError; +pub use crate::ops::TensorOps; diff --git a/crates/kornia-core-ops/src/ops.rs b/crates/kornia-core-ops/src/ops.rs index 05a2d392..01cc4b82 100644 --- a/crates/kornia-core-ops/src/ops.rs +++ b/crates/kornia-core-ops/src/ops.rs @@ -49,7 +49,7 @@ where /// /// ``` /// use kornia_core::{Tensor, CpuAllocator}; - /// use kornia_core_ops::ops::TensorOps; + /// use kornia_core_ops::TensorOps; /// /// let data: [u8; 6] = [1, 1, 1, 1, 1, 1]; /// let t = Tensor::::from_shape_slice([2, 3], &data, CpuAllocator).unwrap(); @@ -80,7 +80,7 @@ where /// /// ``` /// use kornia_core::{Tensor, CpuAllocator}; - /// use kornia_core_ops::ops::TensorOps; + /// use kornia_core_ops::TensorOps; /// /// let data: [f32; 5] = [3., 3., 2., 1., 1.]; /// let t = Tensor::::from_shape_slice([5], &data, CpuAllocator).unwrap(); @@ -110,7 +110,7 @@ where /// /// ``` /// use kornia_core::{Tensor, CpuAllocator}; - /// use kornia_core_ops::ops::TensorOps; + /// use kornia_core_ops::TensorOps; /// /// let data: [f32; 5] = [3., 2., 1., 1., 1.]; /// let t = Tensor::::from_shape_slice([5], &data, CpuAllocator).unwrap();