Skip to main content

conv2d_device

mlx_rs::ops

Function conv2d_device 

Source 
pub fn conv2d_device(
    array: impl AsRef<Array>,
    weight: impl AsRef<Array>,
    stride: impl Into<Option<(i32, i32)>>,
    padding: impl Into<Option<(i32, i32)>>,
    dilation: impl Into<Option<(i32, i32)>>,
    groups: impl Into<Option<i32>>,
    stream: impl AsRef<Stream>,
) -> Result<Array>
👎Deprecated since 0.26.0:

use with_stream or with_device around conv2d

Expand description

Compatibility shim for conv2d.