diff options
| author |  Yu-Cheng Ling <ycling@google.com> | 2018-08-16 12:30:14 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-16 12:33:48 -0700 |
| commit | b09f347b6498edbd0aeeaa1001930e968e80dd49 (patch) | |
| tree | 107119745508ea8a78201e1c4155e4ae5a6a89de /tensorflow/contrib/lite/kernels/register.cc | |
| parent | 1326f33515dac82fbdd7ef502d1df5e96986fc12 (diff) | |
Refactoring: Extract IsEagerOp function to shared component.
PiperOrigin-RevId: 209027338
Diffstat (limited to 'tensorflow/contrib/lite/kernels/register.cc')
| -rw-r--r-- | tensorflow/contrib/lite/kernels/register.cc | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/tensorflow/contrib/lite/kernels/register.cc b/tensorflow/contrib/lite/kernels/register.cc index 6159311910..9681b900b7 100644 --- a/tensorflow/contrib/lite/kernels/register.cc +++ b/tensorflow/contrib/lite/kernels/register.cc @@ -14,6 +14,7 @@ limitations under the License. ==============================================================================*/ #include "tensorflow/contrib/lite/kernels/register.h" +#include "tensorflow/contrib/lite/util.h" namespace tflite { namespace ops { @@ -129,9 +130,7 @@ const TfLiteRegistration* BuiltinOpResolver::FindOp(const char* op, int version) const { // Return the NULL Op for all ops whose name start with "Eager", allowing // the interpreter to delegate their execution. - // TODO(ycling): Refactoring and extract an `IsEagerOp` function into - // `lite:framework` build target. - if (string(op).find("Eager") == 0) { + if (IsEagerOp(op)) { static TfLiteRegistration null_op{ nullptr, nullptr, &UnsupportedTensorFlowOp, nullptr, nullptr, BuiltinOperator_CUSTOM, |