experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add bug for TODO in conv and depthwise conv range check.

Browse Source 
PiperOrigin-RevId: 344377636
Change-Id: Ic955e569f9848417c643f9bc4b43cefd63fa8503
This commit is contained in:
Jaehong Kim 2020-11-26 00:29:18 -08:00 committed by TensorFlower Gardener
parent 1d5f98460a
commit 62c7ae1301
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/lite/kernels/internal/reference/integer_ops/conv.h
View File

@ -101,7 +101,7 @@ inline void ConvPerChannel(
// long as the filter size (filter_y * filter_x * in_channel)
// does not exceed 2^16, which is the case in all the models
// we have seen so far.
// TODO(jianlijianli): Add a check to make sure the
// TODO(b/174275578): Add a check to make sure the
// accumulator depth is smaller than 2^16.
acc += filter_val * (input_val + input_offset);
}

2
tensorflow/lite/kernels/internal/reference/integer_ops/depthwise_conv.h
View File

@ -95,7 +95,7 @@ inline void DepthwiseConvPerChannel(
// long as the filter size (filter_y * filter_x * in_channel)
// does not exceed 2^16, which is the case in all the models
// we have seen so far.
// TODO(jianlijianli): Add a check to make sure the
// TODO(b/174275578): Add a check to make sure the
// accumulator depth is smaller than 2^16.
acc += filter_val * (input_val + input_offset);
}