diff options
| author |  Alan Chiao <alanchiao@google.com> | 2018-05-04 10:31:01 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-04 10:56:20 -0700 |
| commit | a5f44b3519627859fb476a9cad1acc354bfa649f (patch) | |
| tree | 616e06d92971a8452cbb179ece80d569bf244ef8 /tensorflow/contrib/lite/kernels/register.cc | |
| parent | 3db0e545d2460be0392dfcaa304231cd2105648e (diff) | |
Implement neg op
PiperOrigin-RevId: 195435079
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 f91d188ffa..29ea718a96 100644 --- a/tensorflow/contrib/lite/kernels/register.cc +++ b/tensorflow/contrib/lite/kernels/register.cc @@ -81,6 +81,7 @@ TfLiteRegistration* Register_MINIMUM(); TfLiteRegistration* Register_ARG_MAX(); TfLiteRegistration* Register_LESS(); TfLiteRegistration* Register_FLOOR(); +TfLiteRegistration* Register_NEG(); BuiltinOpResolver::BuiltinOpResolver() { AddBuiltin(BuiltinOperator_RELU, Register_RELU()); @@ -143,6 +144,7 @@ BuiltinOpResolver::BuiltinOpResolver() { AddBuiltin(BuiltinOperator_ARG_MAX, Register_ARG_MAX()); AddBuiltin(BuiltinOperator_LESS, Register_LESS()); AddBuiltin(BuiltinOperator_FLOOR, Register_FLOOR()); + AddBuiltin(BuiltinOperator_NEG, Register_NEG()); // TODO(andrewharp, ahentz): Move these somewhere more appropriate so that // custom ops aren't always included by default. |