diff options
| author | Jeremy Johnson <jeremy.johnson@arm.com> | 2023-04-24 09:49:58 +0100 |
|---|---|---|
| committer | Jeremy Johnson <jeremy.johnson@arm.com> | 2023-04-24 09:49:58 +0100 |
| commit | 44eb88db7b601aa18fc7bc77601d6291933bdbb9 (patch) | |
| tree | 66316c1d4619c54ed26d41ce241f335af561e1cd | |
| parent | a793f4645d2c04543572de4d0bc84bf0a3689604 (diff) | |
| download | reference_model-44eb88db7b601aa18fc7bc77601d6291933bdbb9.tar.gz | |
AVG_POOL2D - Fix for stride renaming mistake
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I6deb355998ce88714b41eedc8170acbd7875f519
| -rw-r--r-- | reference_model/src/ops/tensor_ops.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/reference_model/src/ops/tensor_ops.cc b/reference_model/src/ops/tensor_ops.cc index ab3919d..b3845df 100644 --- a/reference_model/src/ops/tensor_ops.cc +++ b/reference_model/src/ops/tensor_ops.cc @@ -593,7 +593,7 @@ int OpAvgPool2d<Dtype, AccDtype>::eval() // calculate 1d height/width div_map (number of elements this pooling window covers) // and outer product to get 2d div_map, then reshape/broadcast to [N, H, W, C] - ETensor1<int32_t> div_map_h = calculate_div_map_1d(in_height, out_height, kernel_y, stride_x, pad_top, pad_bottom); + ETensor1<int32_t> div_map_h = calculate_div_map_1d(in_height, out_height, kernel_y, stride_y, pad_top, pad_bottom); ETensor1<int32_t> div_map_w = calculate_div_map_1d(in_width, out_width, kernel_x, stride_x, pad_left, pad_right); Eigen::array<Eigen::IndexPair<Eigen::Index>, 1> contract_dims = { Eigen::IndexPair<Eigen::Index>(1, 0) }; Eigen::array<Eigen::Index, 4> bcast{ out_batch, 1, 1, out_channels }; |