diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-08 22:49:20 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-09 10:41:25 -0700 |
commit | eac758802e66934a6fde4e23fd92023780a5c075 (patch) | |
tree | bdef3d3c378bafae1f1908d8e3b89523046cd87e /tensorflow/contrib/lite/kernels/register.cc | |
parent | 7bd992b02c0a19ce7aa9c085ab5caa0e00fe2516 (diff) |
Implementation of Slice.
PiperOrigin-RevId: 195926057
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 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. |