Skip to content

GitLab

Commit 487d3751 authored by rocking's avatar rocking
Browse files

Fix bug in example

parent 7beb386f
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
example/51_avgpool3d_bwd/avgpool3d_bwd_common.hpp
View file @ 487d3751
...@@ -77,7 +77,7 @@ bool pool3d_bwd_test(bool do_verification, ...@@ -77,7 +77,7 @@ bool pool3d_bwd_test(bool do_verification,
ck::index_t Do = OutSpatialLength(Di, 0); ck::index_t Do = OutSpatialLength(Di, 0);
ck::index_t Ho = OutSpatialLength(Hi, 1); ck::index_t Ho = OutSpatialLength(Hi, 1);
ck::index_t Wo = OutSpatialLength(Wi, 1); ck::index_t Wo = OutSpatialLength(Wi, 2);
Tensor<DOutDataType> dout(f_host_tensor_descriptor(N, C, Do, Ho, Wo, DOutLayout{})); Tensor<DOutDataType> dout(f_host_tensor_descriptor(N, C, Do, Ho, Wo, DOutLayout{}));
Tensor<DInDataType> din_dev(f_host_tensor_descriptor(N, C, Di, Hi, Wi, DInLayout{})); Tensor<DInDataType> din_dev(f_host_tensor_descriptor(N, C, Di, Hi, Wi, DInLayout{}));
......
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