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 | |
| 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
| -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; |