Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
composable_kernel
Commits
01aeb901
Commit
01aeb901
authored
Jul 21, 2023
by
rocking
Browse files
calculate Do Ho Wo according to the dilation
parent
487d3751
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
1 deletion
+3
-1
example/51_avgpool3d_bwd/avgpool3d_bwd_common.hpp
example/51_avgpool3d_bwd/avgpool3d_bwd_common.hpp
+3
-1
No files found.
example/51_avgpool3d_bwd/avgpool3d_bwd_common.hpp
View file @
01aeb901
...
@@ -72,7 +72,9 @@ bool pool3d_bwd_test(bool do_verification,
...
@@ -72,7 +72,9 @@ bool pool3d_bwd_test(bool do_verification,
ck
::
index_t
right_pad
=
dinput_right_pads
[
index
];
ck
::
index_t
right_pad
=
dinput_right_pads
[
index
];
ck
::
index_t
window_len
=
window_lengths
[
index
];
ck
::
index_t
window_len
=
window_lengths
[
index
];
ck
::
index_t
stride
=
window_strides
[
index
];
ck
::
index_t
stride
=
window_strides
[
index
];
return
(
InSpatialLength
+
left_pad
+
right_pad
-
window_len
)
/
stride
+
1
;
ck
::
index_t
dilation
=
window_dilations
[
index
];
ck
::
index_t
eff
=
(
window_len
-
1
)
*
dilation
+
1
;
return
(
InSpatialLength
+
left_pad
+
right_pad
-
eff
)
/
stride
+
1
;
};
};
ck
::
index_t
Do
=
OutSpatialLength
(
Di
,
0
);
ck
::
index_t
Do
=
OutSpatialLength
(
Di
,
0
);
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment