Fix ndims argument value to convolve*_nn C API calls

Signed-off-by: Pradeep Garigipati <pradeep.garigipati@gmail.com>

Author: 9prady9

src/ml/mod.rs

@@ -76,11 +76,11 @@ where
             &mut temp as *mut af_array,
             signal.get(),
             filter.get(),
-            strides.ndims() as c_uint,
+            4,
             strides.get().as_ptr() as *const dim_t,
-            padding.ndims() as c_uint,
+            4,
             padding.get().as_ptr() as *const dim_t,
-            dilation.ndims() as c_uint,
+            4,
             dilation.get().as_ptr() as *const dim_t,
         );
         HANDLE_ERROR(AfError::from(err_val));
@@ -126,11 +126,11 @@ where
             original_signal.get(),
             original_filter.get(),
             convolved_output.get(),
-            strides.ndims() as c_uint,
+            4,
             strides.get().as_ptr() as *const dim_t,
-            padding.ndims() as c_uint,
+            4,
             padding.get().as_ptr() as *const dim_t,
-            dilation.ndims() as c_uint,
+            4,
             dilation.get().as_ptr() as *const dim_t,
             grad_type as c_uint,
         );