diff options
author | Avijit <Avijit.Chakraborty@intel.com> | 2018-08-15 17:00:22 -0700 |
---|---|---|
committer | Avijit <Avijit.Chakraborty@intel.com> | 2018-08-15 17:00:22 -0700 |
commit | bc6be507c71046dfc889a90e3949a903d5d1e6eb (patch) | |
tree | 84557e7bb7798e3d418a619c8452aa7baf78f255 /tensorflow/contrib/lite/kernels/register.cc | |
parent | 9523a98466d16cf01fc76a67b489f1124cf626ac (diff) | |
parent | d2875ea71373d05c645587a83dd870fa8a0ec070 (diff) |
Merge remote-tracking branch 'upstream/master'
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, |