aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/kernels/register.cc
diff options
context:
space:
mode:
authorGravatar Avijit <Avijit.Chakraborty@intel.com>2018-08-15 17:00:22 -0700
committerGravatar Avijit <Avijit.Chakraborty@intel.com>2018-08-15 17:00:22 -0700
commitbc6be507c71046dfc889a90e3949a903d5d1e6eb (patch)
tree84557e7bb7798e3d418a619c8452aa7baf78f255 /tensorflow/contrib/lite/kernels/register.cc
parent9523a98466d16cf01fc76a67b489f1124cf626ac (diff)
parentd2875ea71373d05c645587a83dd870fa8a0ec070 (diff)
Merge remote-tracking branch 'upstream/master'
Diffstat (limited to 'tensorflow/contrib/lite/kernels/register.cc')
-rw-r--r--tensorflow/contrib/lite/kernels/register.cc6
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,