diff options
author | 2018-09-11 21:18:05 -0700 | |
---|---|---|
committer | 2018-09-11 21:22:09 -0700 | |
commit | 6a21e1386e3e68cf752af861b9b1b950bda8a130 (patch) | |
tree | 0ccc9df1c589ffb059b707c58d8c7c094f6f9de0 /tensorflow/contrib/lite/kernels/register.cc | |
parent | cadd6b42bf6b01c2668420463b0986acd7fd9009 (diff) |
Implementation of square.
PiperOrigin-RevId: 212577288
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 c66959fdf4..14296d3a9f 100644 --- a/tensorflow/contrib/lite/kernels/register.cc +++ b/tensorflow/contrib/lite/kernels/register.cc @@ -118,6 +118,7 @@ TfLiteRegistration* Register_LOGICAL_AND(); TfLiteRegistration* Register_LOGICAL_NOT(); TfLiteRegistration* Register_UNPACK(); TfLiteRegistration* Register_FLOOR_DIV(); +TfLiteRegistration* Register_SQUARE(); TfLiteStatus UnsupportedTensorFlowOp(TfLiteContext* context, TfLiteNode* node) { context->ReportError( @@ -243,6 +244,7 @@ BuiltinOpResolver::BuiltinOpResolver() { AddBuiltin(BuiltinOperator_LOGICAL_NOT, Register_LOGICAL_NOT()); AddBuiltin(BuiltinOperator_UNPACK, Register_UNPACK()); AddBuiltin(BuiltinOperator_FLOOR_DIV, Register_FLOOR_DIV()); + AddBuiltin(BuiltinOperator_SQUARE, Register_SQUARE()); // TODO(andrewharp, ahentz): Move these somewhere more appropriate so that // custom ops aren't always included by default. |