Add tensor_buffer() & tensor_buffer_mut() for Interpreter

NarrativeApp · Dec 26, 2019 · d70cda6 · d70cda6
1 parent 53e2002
commit d70cda6
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/interpreter/mod.rs b/src/interpreter/mod.rs
@@ -397,6 +397,18 @@ where
             slice::from_raw_parts_mut(inner.data.raw as *mut T, inner.bytes / size_of::<T>())
         })
     }
+
+    pub fn tensor_buffer(&self, tensor_index: TensorIndex) -> Fallible<&[u8]> {
+        let inner = self.tensor_inner(tensor_index)?;
+
+        Ok(unsafe { slice::from_raw_parts(inner.data.raw_const as *mut u8, inner.bytes) })
+    }
+
+    pub fn tensor_buffer_mut(&mut self, tensor_index: TensorIndex) -> Fallible<&mut [u8]> {
+        let inner = self.tensor_inner(tensor_index)?;
+
+        Ok(unsafe { slice::from_raw_parts_mut(inner.data.raw as *mut u8, inner.bytes) })
+    }
 }
 
 #[cfg(test)]