diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-19 09:49:17 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-19 09:54:05 -0700 |
commit | 5fc2bdd2d5f624a6bad9e83b992029e3799ab64e (patch) | |
tree | 576404472bd26801fe2fce6f6385669aef1f42e4 /tensorflow/contrib/lite/kernels/register.cc | |
parent | c532c3f319c72074e6fb8cb10c6d05a3839bcc0a (diff) |
Implement TFLite sqrt/rsqrt unary operators
PiperOrigin-RevId: 201191877
Diffstat (limited to 'tensorflow/contrib/lite/kernels/register.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/register.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/register.cc b/tensorflow/contrib/lite/kernels/register.cc index b893e40fe3..07a7ee9115 100644 --- a/tensorflow/contrib/lite/kernels/register.cc +++ b/tensorflow/contrib/lite/kernels/register.cc @@ -98,6 +98,8 @@ TfLiteRegistration* Register_EXPAND_DIMS(); TfLiteRegistration* Register_SPARSE_TO_DENSE(); TfLiteRegistration* Register_EQUAL(); TfLiteRegistration* Register_NOT_EQUAL(); +TfLiteRegistration* Register_SQRT(); +TfLiteRegistration* Register_RSQRT(); BuiltinOpResolver::BuiltinOpResolver() { AddBuiltin(BuiltinOperator_RELU, Register_RELU()); @@ -177,6 +179,8 @@ BuiltinOpResolver::BuiltinOpResolver() { AddBuiltin(BuiltinOperator_SPARSE_TO_DENSE, Register_SPARSE_TO_DENSE()); AddBuiltin(BuiltinOperator_EQUAL, Register_EQUAL()); AddBuiltin(BuiltinOperator_NOT_EQUAL, Register_NOT_EQUAL()); + AddBuiltin(BuiltinOperator_SQRT, Register_SQRT()); + AddBuiltin(BuiltinOperator_RSQRT, Register_RSQRT()); // TODO(andrewharp, ahentz): Move these somewhere more appropriate so that // custom ops aren't always included by default. |