diff options
| -rw-r--r-- | reference_model/src/ops/tensor_ops.cc | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/reference_model/src/ops/tensor_ops.cc b/reference_model/src/ops/tensor_ops.cc index 96fd41c..acdeeeb 100644 --- a/reference_model/src/ops/tensor_ops.cc +++ b/reference_model/src/ops/tensor_ops.cc @@ -1233,9 +1233,10 @@ int OpDepthwiseConv2d<InDtype, WeightDtype, OutDtype>::eval() for (int fw = 0; fw < f_width; fw++) { // Perform multiplication in AccEigenType then cast to OutEigenType - this->output->getTensor()(ob, oh, ow, ic * f_multiplier + cm) += (OutEigenType)( - (AccEigenType)input_extract_patches(ob, fh, fw, ow * out_height + oh, ic) * - (AccEigenType)weight_val(fh, fw, ic, cm)); + this->output->getTensor()(ob, oh, ow, ic * f_multiplier + cm) += + (OutEigenType)((AccEigenType)input_extract_patches(ob, fh, fw, ow * out_height + oh, + ic) * + (AccEigenType)weight_val(fh, fw, ic, cm)); } } } |