pub struct TensorMemoryRequirements {
pub total_size: usize,
pub max_tensor_size: usize,
pub tensor_count: usize,
pub quantized_tensor_count: usize,
pub non_quantized_tensor_count: usize,
pub recommended_buffer_size: usize,
}Expand description
Memory requirements for reading tensors
Fields§
§total_size: usizeTotal size of all tensors
max_tensor_size: usizeSize of the largest tensor
tensor_count: usizeTotal number of tensors
quantized_tensor_count: usizeNumber of quantized tensors
non_quantized_tensor_count: usizeNumber of non-quantized tensors
recommended_buffer_size: usizeRecommended buffer size for reading
Implementations§
Source§impl TensorMemoryRequirements
impl TensorMemoryRequirements
Sourcepub fn average_tensor_size(&self) -> usize
pub fn average_tensor_size(&self) -> usize
Get the average tensor size
Sourcepub fn is_reasonable(&self, available_memory: usize) -> bool
pub fn is_reasonable(&self, available_memory: usize) -> bool
Check if memory requirements are reasonable
Trait Implementations§
Source§impl Clone for TensorMemoryRequirements
impl Clone for TensorMemoryRequirements
Source§impl Debug for TensorMemoryRequirements
impl Debug for TensorMemoryRequirements
Source§impl Display for TensorMemoryRequirements
impl Display for TensorMemoryRequirements
impl Eq for TensorMemoryRequirements
Source§impl PartialEq for TensorMemoryRequirements
impl PartialEq for TensorMemoryRequirements
impl StructuralPartialEq for TensorMemoryRequirements
Auto Trait Implementations§
impl Freeze for TensorMemoryRequirements
impl RefUnwindSafe for TensorMemoryRequirements
impl Send for TensorMemoryRequirements
impl Sync for TensorMemoryRequirements
impl Unpin for TensorMemoryRequirements
impl UnsafeUnpin for TensorMemoryRequirements
impl UnwindSafe for TensorMemoryRequirements
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more