diff options
Diffstat (limited to 'reference_model/src/ops')
-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; |