diff options
Diffstat (limited to 'candle-core/src/layout.rs')
| -rw-r--r-- | candle-core/src/layout.rs | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/candle-core/src/layout.rs b/candle-core/src/layout.rs index bf346cf2..e6824b29 100644 --- a/candle-core/src/layout.rs +++ b/candle-core/src/layout.rs @@ -70,7 +70,7 @@ impl Layout { self.shape.is_fortran_contiguous(&self.stride) } - pub(crate) fn narrow(&self, dim: usize, start: usize, len: usize) -> Result<Self> { + pub fn narrow(&self, dim: usize, start: usize, len: usize) -> Result<Self> { let dims = self.shape().dims(); if dim >= dims.len() { Err(Error::DimOutOfRange { @@ -99,7 +99,7 @@ impl Layout { }) } - pub(crate) fn transpose(&self, dim1: usize, dim2: usize) -> Result<Self> { + pub fn transpose(&self, dim1: usize, dim2: usize) -> Result<Self> { let rank = self.shape.rank(); if rank <= dim1 || rank <= dim2 { Err(Error::UnexpectedNumberOfDims { @@ -120,7 +120,7 @@ impl Layout { }) } - pub(crate) fn permute(&self, idxs: &[usize]) -> Result<Self> { + pub fn permute(&self, idxs: &[usize]) -> Result<Self> { let is_permutation = idxs.len() == self.shape.rank() && (0..idxs.len()).all(|i| idxs.contains(&i)); if !is_permutation { |