From 44eb88db7b601aa18fc7bc77601d6291933bdbb9 Mon Sep 17 00:00:00 2001 From: Jeremy Johnson Date: Mon, 24 Apr 2023 09:49:58 +0100 Subject: AVG_POOL2D - Fix for stride renaming mistake Signed-off-by: Jeremy Johnson Change-Id: I6deb355998ce88714b41eedc8170acbd7875f519 --- reference_model/src/ops/tensor_ops.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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::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 div_map_h = calculate_div_map_1d(in_height, out_height, kernel_y, stride_x, pad_top, pad_bottom); + ETensor1 div_map_h = calculate_div_map_1d(in_height, out_height, kernel_y, stride_y, pad_top, pad_bottom); ETensor1 div_map_w = calculate_div_map_1d(in_width, out_width, kernel_x, stride_x, pad_left, pad_right); Eigen::array, 1> contract_dims = { Eigen::IndexPair(1, 0) }; Eigen::array bcast{ out_batch, 1, 1, out_channels }; -- cgit v1.2.1