diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-15 23:32:33 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-15 23:35:03 -0700 |
commit | 990e1f218c7180b2ebf407b8ec06d59936e9cc12 (patch) | |
tree | 0386a3f0afcf8468ddaf0d6e9616b6b3f1e82df5 /tensorflow/contrib/lite/kernels/register.cc | |
parent | df4ff7833725452c4ede1bf58b7523bafff3ecef (diff) |
TFLite Custom op for object detection postprocessing.
PiperOrigin-RevId: 200820561
Diffstat (limited to 'tensorflow/contrib/lite/kernels/register.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/register.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/register.cc b/tensorflow/contrib/lite/kernels/register.cc index 7bb28d4de7..98f7250a40 100644 --- a/tensorflow/contrib/lite/kernels/register.cc +++ b/tensorflow/contrib/lite/kernels/register.cc @@ -22,6 +22,7 @@ namespace custom { TfLiteRegistration* Register_AUDIO_SPECTROGRAM(); TfLiteRegistration* Register_MFCC(); +TfLiteRegistration* Register_SSD_POSTPROCESS(); } // namespace custom @@ -180,6 +181,8 @@ BuiltinOpResolver::BuiltinOpResolver() { AddCustom("Mfcc", tflite::ops::custom::Register_MFCC()); AddCustom("AudioSpectrogram", tflite::ops::custom::Register_AUDIO_SPECTROGRAM()); + AddCustom("TFLite_SSD_PostProcess", + tflite::ops::custom::Register_SSD_POSTPROCESS()); } } // namespace builtin |