Movatterモバイル変換


[0]ホーム

URL:


Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Sign up
Appearance settings

Commit140ee83

Browse files
committed
Fix ndims argument value to convolve*_nn C API calls
Signed-off-by: Pradeep Garigipati <pradeep.garigipati@gmail.com>
1 parent2bb1b3e commit140ee83

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

‎src/ml/mod.rs

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -76,11 +76,11 @@ where
7676
&mut tempas*mutaf_array,
7777
signal.get(),
7878
filter.get(),
79-
strides.ndims()asc_uint,
79+
4,
8080
strides.get().as_ptr()as*constdim_t,
81-
padding.ndims()asc_uint,
81+
4,
8282
padding.get().as_ptr()as*constdim_t,
83-
dilation.ndims()asc_uint,
83+
4,
8484
dilation.get().as_ptr()as*constdim_t,
8585
);
8686
HANDLE_ERROR(AfError::from(err_val));
@@ -126,11 +126,11 @@ where
126126
original_signal.get(),
127127
original_filter.get(),
128128
convolved_output.get(),
129-
strides.ndims()asc_uint,
129+
4,
130130
strides.get().as_ptr()as*constdim_t,
131-
padding.ndims()asc_uint,
131+
4,
132132
padding.get().as_ptr()as*constdim_t,
133-
dilation.ndims()asc_uint,
133+
4,
134134
dilation.get().as_ptr()as*constdim_t,
135135
grad_typeasc_uint,
136136
);

0 commit comments

Comments
 (0)

[8]ページ先頭

©2009-2025 Movatter.jp