diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-08-28 06:56:23 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-28 07:01:31 -0700 |
| commit | 5656c3db01c8d98758c0edeb6934dbd4698f39d1 (patch) | |
| tree | 3470b6c2fd4d070bfe08d341220702011859d175 /tensorflow/contrib/lite/kernels/register.cc | |
| parent | de1696e9a818646fe6f200db42b150f1b7141900 (diff) | |
Implementation of floor_div.
PiperOrigin-RevId: 210533721
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 175dfab210..7b859dc332 100644 --- a/tensorflow/contrib/lite/kernels/register.cc +++ b/tensorflow/contrib/lite/kernels/register.cc @@ -115,6 +115,7 @@ TfLiteRegistration* Register_LOGICAL_OR(); TfLiteRegistration* Register_LOGICAL_AND(); TfLiteRegistration* Register_LOGICAL_NOT(); TfLiteRegistration* Register_UNPACK(); +TfLiteRegistration* Register_FLOOR_DIV(); TfLiteStatus UnsupportedTensorFlowOp(TfLiteContext* context, TfLiteNode* node) { context->ReportError( @@ -239,6 +240,7 @@ BuiltinOpResolver::BuiltinOpResolver() { AddBuiltin(BuiltinOperator_LOGICAL_AND, Register_LOGICAL_AND()); AddBuiltin(BuiltinOperator_LOGICAL_NOT, Register_LOGICAL_NOT()); AddBuiltin(BuiltinOperator_UNPACK, Register_UNPACK()); + AddBuiltin(BuiltinOperator_FLOOR_DIV, Register_FLOOR_DIV()); // TODO(andrewharp, ahentz): Move these somewhere more appropriate so that // custom ops aren't always included by default. |