| commit | c648eb8d93676d8b8af8efc5a0c8d9746890e750 | [log] [tgz] |
|---|---|---|
| author | Nat Jeffries <njeff@google.com> | Thu Apr 16 19:25:36 2020 -0700 |
| committer | TensorFlower Gardener <gardener@tensorflow.org> | Thu Apr 16 19:28:47 2020 -0700 |
| tree | e546b71ef1f4295ee5d53e1cecf302e9e2ae39c7 | |
| parent | d677fa2407677acdf710f7396acd31c706f13bd3 [diff] |
Fix xtensa optimized conv and depthwise conv for models with 32 channels. PiperOrigin-RevId: 306968609 Change-Id: Ibb8f26b55abb19aefbf561544113df482b37ee58
diff --git a/tensorflow/lite/micro/kernels/xtensa_hifimini/conv.cc b/tensorflow/lite/micro/kernels/xtensa_hifimini/conv.cc index a1de5ef..c8647d1 100644 --- a/tensorflow/lite/micro/kernels/xtensa_hifimini/conv.cc +++ b/tensorflow/lite/micro/kernels/xtensa_hifimini/conv.cc
@@ -177,7 +177,7 @@ constexpr int kFilterTensor = 1; constexpr int kBiasTensor = 2; constexpr int kOutputTensor = 0; -constexpr int kMaxChannels = 8; +constexpr int kMaxChannels = 32; // Conv is quantized along dimension 0: // https://www.tensorflow.org/lite/performance/quantization_spec
diff --git a/tensorflow/lite/micro/kernels/xtensa_hifimini/depthwise_conv.cc b/tensorflow/lite/micro/kernels/xtensa_hifimini/depthwise_conv.cc index 3760dd7..692ab89 100644 --- a/tensorflow/lite/micro/kernels/xtensa_hifimini/depthwise_conv.cc +++ b/tensorflow/lite/micro/kernels/xtensa_hifimini/depthwise_conv.cc
@@ -180,7 +180,7 @@ constexpr int kFilterTensor = 1; constexpr int kBiasTensor = 2; constexpr int kOutputTensor = 0; -constexpr int kMaxChannels = 8; +constexpr int kMaxChannels = 32; // Depthwise conv is quantized along dimension 3: // https://www.tensorflow.org/lite/performance/quantization_spec