Skip to content

GitLab

Commit 814fe1bc authored by rocking's avatar rocking
Browse files

Extract variable

parent a465273a
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
profiler/include/profiler/profile_pool2d_fwd_impl.hpp
View file @ 814fe1bc
......@@ -148,9 +148,9 @@ bool profile_pool2d_fwd_impl(int do_verification,
{C * Hi * Wi, 1, Wi * C, C},
{C * Ho * Wo, 1, Wo * C, C},
{C * Ho * Wo, 1, Wo * C, C},
{N, C, Hi, Wi},
in_length,
window_spatial_lengths,
{N, C, Ho, Wo},
out_length,
window_strides,
input_left_pads,
input_right_pads,
......
profiler/include/profiler/profile_pool3d_fwd_impl.hpp
View file @ 814fe1bc
......@@ -153,9 +153,9 @@ bool profile_pool3d_fwd_impl(int do_verification,
{Di * C * Hi * Wi, 1, C * Hi * Wi, Wi * C, C},
{Do * C * Ho * Wo, 1, C * Ho * Wo, Wo * C, C},
{Do * C * Ho * Wo, 1, C * Ho * Wo, Wo * C, C},
{N, C, Di, Hi, Wi},
in_length,
window_spatial_lengths,
{N, C, Do, Ho, Wo},
out_length,
window_strides,
input_left_pads,
input_right_pads,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment