Making ToDtypeKernel pub (#868)

coreylowman · Sep 14, 2023 · 2e5116c · 2e5116c
1 parent 9c528eb
commit 2e5116c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/tensor_ops/mod.rs b/src/tensor_ops/mod.rs
@@ -269,12 +269,11 @@ pub use stddev_to::StddevTo;
 pub use sub::{sub, TrySub};
 pub use sum_to::SumTo;
 pub use tanh::tanh;
-pub use to_dtype::to_dtype;
+pub use to_dtype::{to_dtype, ToDtypeKernel};
 pub use tri::{lower_tri, upper_tri};
 pub use upscale2d::{Bilinear, GenericUpscale2D, NearestNeighbor, TryUpscale2D, UpscaleMethod};
 pub use var_to::VarTo;
 
-pub(crate) use to_dtype::ToDtypeKernel;
 pub(crate) use upscale2d::Upscale2DKernel;
 
 #[cfg(feature = "nightly")]