CacheReturn

gradients::devices::cache

Trait CacheReturn 

Source 
pub trait CacheReturn<P>: GraphReturn
where
    P: CacheType,
{
    // Required method
    fn cache(&self) -> RefMut<'_, Cache<P>>;
}
Expand description

This trait makes a device’s Cache accessible and is implemented for all compute devices.

Required Methods§

Source

fn cache(&self) -> RefMut<'_, Cache<P>>

Returns a device specific Cache.

Implementors§