diff options
| author |  Alan Chiao <alanchiao@google.com> | 2018-09-04 20:22:55 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-04 20:26:56 -0700 |
| commit | e9332539bea372f6dbe6ef185f9d8b1f3b6e1fe2 (patch) | |
| tree | c7d5490402934bc34b6ad76a89ac14233e91a848 /tensorflow/contrib/lite/kernels/register.cc | |
| parent | ecb6bc19e0cdbd2f2e98de909b4f3b8ca9fd7ab1 (diff) | |
Relu1 custom op.
This is implemented as custom op instead of builtin op because Relu1 is not
supported in Tensorflow and not commonly used.
PiperOrigin-RevId: 211571619
Diffstat (limited to 'tensorflow/contrib/lite/kernels/register.cc')
| -rw-r--r-- | tensorflow/contrib/lite/kernels/register.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/register.cc b/tensorflow/contrib/lite/kernels/register.cc index 188015f43c..c66959fdf4 100644 --- a/tensorflow/contrib/lite/kernels/register.cc +++ b/tensorflow/contrib/lite/kernels/register.cc @@ -25,6 +25,7 @@ TfLiteRegistration* Register_AUDIO_SPECTROGRAM(); TfLiteRegistration* Register_LAYER_NORM_LSTM(); TfLiteRegistration* Register_MFCC(); TfLiteRegistration* Register_DETECTION_POSTPROCESS(); +TfLiteRegistration* Register_RELU_1(); } // namespace custom @@ -249,6 +250,7 @@ BuiltinOpResolver::BuiltinOpResolver() { AddCustom("AudioSpectrogram", tflite::ops::custom::Register_AUDIO_SPECTROGRAM()); AddCustom("LayerNormLstm", tflite::ops::custom::Register_LAYER_NORM_LSTM()); + AddCustom("Relu1", tflite::ops::custom::Register_RELU_1()); AddCustom("TFLite_Detection_PostProcess", tflite::ops::custom::Register_DETECTION_POSTPROCESS()); } |