aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/kernels/register.cc
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-05-08 22:49:20 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-05-09 10:41:25 -0700
commiteac758802e66934a6fde4e23fd92023780a5c075 (patch)
treebdef3d3c378bafae1f1908d8e3b89523046cd87e /tensorflow/contrib/lite/kernels/register.cc
parent7bd992b02c0a19ce7aa9c085ab5caa0e00fe2516 (diff)
Implementation of Slice.
PiperOrigin-RevId: 195926057
Diffstat (limited to 'tensorflow/contrib/lite/kernels/register.cc')
-rw-r--r--tensorflow/contrib/lite/kernels/register.cc2
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/register.cc b/tensorflow/contrib/lite/kernels/register.cc
index 5df35aac62..4544f2d292 100644
--- a/tensorflow/contrib/lite/kernels/register.cc
+++ b/tensorflow/contrib/lite/kernels/register.cc
@@ -87,6 +87,7 @@ TfLiteRegistration* Register_LESS_EQUAL();
TfLiteRegistration* Register_FLOOR();
TfLiteRegistration* Register_NEG();
TfLiteRegistration* Register_SELECT();
+TfLiteRegistration* Register_SLICE();
BuiltinOpResolver::BuiltinOpResolver() {
AddBuiltin(BuiltinOperator_RELU, Register_RELU());
@@ -155,6 +156,7 @@ BuiltinOpResolver::BuiltinOpResolver() {
AddBuiltin(BuiltinOperator_FLOOR, Register_FLOOR());
AddBuiltin(BuiltinOperator_NEG, Register_NEG());
AddBuiltin(BuiltinOperator_SELECT, Register_SELECT());
+ AddBuiltin(BuiltinOperator_SLICE, Register_SLICE());
// TODO(andrewharp, ahentz): Move these somewhere more appropriate so that
// custom ops aren't always included by default.