diff options
Diffstat (limited to 'pseudocode/operators')
| -rw-r--r-- | pseudocode/operators/CAST.tosac | 4 | ||||
| -rw-r--r-- | pseudocode/operators/DEPTHWISE_CONV2D.tosac | 4 |
2 files changed, 4 insertions, 4 deletions
diff --git a/pseudocode/operators/CAST.tosac b/pseudocode/operators/CAST.tosac index 4e2dca3..b8fdea9 100644 --- a/pseudocode/operators/CAST.tosac +++ b/pseudocode/operators/CAST.tosac @@ -28,11 +28,11 @@ for_each(index in shape) { out = (in) ? 1 : 0; } else if (is_floating_point<in_t>()) { out = truncate<out_t>(apply_clip_s<i32_t>(round_to_nearest_int(in), minimum_s<out_t>(), maximum_s<out_t>())); - } else if (sizeof(out_t) >= sizeof(in_t)) { + } else if (sizeof<out_t>() >= sizeof<in_t>()) { out = sign_extend<out_t>(in); } else { out = truncate<out_t>(in); } } - tensor_write<out_t>(output, shape, index, out) + tensor_write<out_t>(output, shape, index, out); } diff --git a/pseudocode/operators/DEPTHWISE_CONV2D.tosac b/pseudocode/operators/DEPTHWISE_CONV2D.tosac index 6cbcc6a..a473375 100644 --- a/pseudocode/operators/DEPTHWISE_CONV2D.tosac +++ b/pseudocode/operators/DEPTHWISE_CONV2D.tosac @@ -30,8 +30,8 @@ for_each(0 <= n < N, 0 <= oy < OH, 0 <= ox < OW, 0 <= c < C, 0 <= m < M) { out_t weight = static_cast<out_t>(tensor_read<weight_t>(weight, [KH,KW,C,M], [ky,kx,c,m])); - value = apply_sub_s<out_t>(value, static_cast<out_t>input_zp); - weight = apply_sub_s<out_t>(weight, static_cast<out_t>weight_zp); + value = apply_sub_s<out_t>(value, static_cast<out_t>(input_zp)); + weight = apply_sub_s<out_t>(weight, static_cast<out_t>(weight_zp)); acc = apply_add_s<out_t>(acc, apply_mul_s<out_t>(value, weight)); } } |