diff options
author | Eric Kunze <eric.kunze@arm.com> | 2022-05-16 17:34:40 -0700 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2022-05-16 17:35:38 -0700 |
commit | 7edb34c02614c48cc9e535c39198711d6692127d (patch) | |
tree | f700aecea980962855c36ca983683b31b8aff764 /reference_model | |
parent | 67a91550dc7691e1c44b1e1f075645ce16b65c3e (diff) | |
download | reference_model-7edb34c02614c48cc9e535c39198711d6692127d.tar.gz |
Initialize accumulator with bias value for CONV3D
Bias was mistakenly unused previously.
Signed-off-by: Eric Kunze <eric.kunze@arm.com>
Change-Id: If776e923f7b3cd442c7f147fcbd22a214c37fd36
Diffstat (limited to 'reference_model')
-rw-r--r-- | reference_model/src/ops/tensor_ops.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/reference_model/src/ops/tensor_ops.cc b/reference_model/src/ops/tensor_ops.cc index 862c640..732480c 100644 --- a/reference_model/src/ops/tensor_ops.cc +++ b/reference_model/src/ops/tensor_ops.cc @@ -890,7 +890,8 @@ int OpConv3d<InDtype, WeightDtype>::eval() { for (int oc = 0; oc < out_channels; oc++) { - acc = 0; + // Initialize accumulator with bias value + acc = this->output->getTensor()(ob, od, oh, ow, oc); for (int fd = 0; fd < f_depth; fd++) { d_idx = od * stride_d + fd * dilation_d; |