diff options
author | Giorgio Arena <giorgio.arena@arm.com> | 2021-10-12 16:00:40 +0100 |
---|---|---|
committer | Sheri Zhang <sheri.zhang@arm.com> | 2021-10-12 16:56:30 +0000 |
commit | 5c002ec70aa20569d44a3e4c5bbcf53135364e7b (patch) | |
tree | cf8d3fd52318f3453ec7b14b627a9f3d03825406 /tests/validation/reference/Conv3D.cpp | |
parent | ded3663274db0e4359461659fb3c813792df16e3 (diff) | |
download | ComputeLibrary-5c002ec70aa20569d44a3e4c5bbcf53135364e7b.tar.gz |
Add missing common functionalities for DirectConv3D
Partially resolves COMPMID-4660
Signed-off-by: Giorgio Arena <giorgio.arena@arm.com>
Change-Id: Iaa659fd9c0ce364e491b04e5ccd1620d69aeca61
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6417
Reviewed-by: Sheri Zhang <sheri.zhang@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/validation/reference/Conv3D.cpp')
-rw-r--r-- | tests/validation/reference/Conv3D.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/validation/reference/Conv3D.cpp b/tests/validation/reference/Conv3D.cpp index 4b0f2b0930..ad61105b36 100644 --- a/tests/validation/reference/Conv3D.cpp +++ b/tests/validation/reference/Conv3D.cpp @@ -141,13 +141,13 @@ SimpleTensor<T> conv3d(const SimpleTensor<T> &src, const SimpleTensor<T> &weight { for(unsigned int z_out = 0; z_out < dst_depth; ++z_out) { - const int z_start = (z_out * stride_z) - pad_left; + const int z_start = (z_out * stride_z) - pad_front; for(unsigned int y_out = 0; y_out < dst_height; ++y_out) { const int y_start = (y_out * stride_y) - pad_top; for(unsigned int x_out = 0; x_out < dst_width; ++x_out) { - const int x_start = (x_out * stride_x) - pad_front; + const int x_start = (x_out * stride_x) - pad_left; for(unsigned int ch_out = 0; ch_out < dst_channels; ++ch_out) { T weighted_value = calculate_conv3d<T>(src, weights, conv3d_info.dilation, batch, z_start, |