diff options
| author |  Yu-Cheng Ling <ycling@google.com> | 2018-08-14 16:47:41 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-14 16:50:59 -0700 |
| commit | 1200245073e531c11f56bdf6581dc9e8df8e0597 (patch) | |
| tree | e64f84c018e32a464a323031e7ed741b8581dc17 /tensorflow/contrib/lite/kernels/register.cc | |
| parent | a05c7b0e0953bf76ccf71ffbff2076150b1fe709 (diff) | |
Fix the Eager prefix constant.
PiperOrigin-RevId: 208738888
Diffstat (limited to 'tensorflow/contrib/lite/kernels/register.cc')
| -rw-r--r-- | tensorflow/contrib/lite/kernels/register.cc | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/kernels/register.cc b/tensorflow/contrib/lite/kernels/register.cc index 8d2c108116..6159311910 100644 --- a/tensorflow/contrib/lite/kernels/register.cc +++ b/tensorflow/contrib/lite/kernels/register.cc @@ -127,9 +127,11 @@ const TfLiteRegistration* BuiltinOpResolver::FindOp(tflite::BuiltinOperator op, const TfLiteRegistration* BuiltinOpResolver::FindOp(const char* op, int version) const { - // Return the NULL Op for all ops whose name start with "Eager:", allowing + // Return the NULL Op for all ops whose name start with "Eager", allowing // the interpreter to delegate their execution. - if (string(op).find("Eager:") == 0) { + // TODO(ycling): Refactoring and extract an `IsEagerOp` function into + // `lite:framework` build target. + if (string(op).find("Eager") == 0) { static TfLiteRegistration null_op{ nullptr, nullptr, &UnsupportedTensorFlowOp, nullptr, nullptr, BuiltinOperator_CUSTOM, |