Skip to content

GitLab

"examples/community/gluegen.py" did not exist on "86ecd4b795f865b5b615b8c54991c177bb3dbef5"
Commit a014aa09 authored by rocking's avatar rocking
Browse files

Fix dimension

parent 16595e96
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
include/ck/tensor_operation/gpu/device/impl/device_pool2d_fwd_nhwc_nhwc.hpp
View file @ a014aa09
......@@ -30,7 +30,7 @@ template <typename InDataType,
ck::index_t ReduceKThreadSliceSize,
ck::index_t InSrcOutDstVectorSize>
struct DevicePool2dFwd_Input_N_Hi_Wi_C_Output_N_Ho_Wo_C
: public DevicePoolFwd<5, 2, ReduceOpId, OuputIndex>
: public DevicePoolFwd<4, 2, ReduceOpId, OuputIndex>
{
static constexpr auto I0 = Number<0>{};
static constexpr auto I1 = Number<1>{};
......@@ -39,7 +39,7 @@ struct DevicePool2dFwd_Input_N_Hi_Wi_C_Output_N_Ho_Wo_C
static constexpr auto I4 = Number<4>{};
static constexpr auto I5 = Number<5>{};
static constexpr index_t InOutRank = 5;
static constexpr index_t InOutRank = 4;
static constexpr index_t WindowRank = 2;
using IndexDataType = int32_t;
......
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