diff options
Diffstat (limited to 'tensorflow/contrib/lite/util.cc')
-rw-r--r-- | tensorflow/contrib/lite/util.cc | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/contrib/lite/util.cc b/tensorflow/contrib/lite/util.cc index 6aa35b5227..7950653da9 100644 --- a/tensorflow/contrib/lite/util.cc +++ b/tensorflow/contrib/lite/util.cc @@ -18,9 +18,9 @@ limitations under the License. namespace tflite { -bool IsFlexOp(const char* custom_name) { - return custom_name && strncmp(custom_name, kFlexCustomCodePrefix, - strlen(kFlexCustomCodePrefix)) == 0; +bool IsEagerOp(const char* custom_name) { + return custom_name && strncmp(custom_name, kEagerCustomCodePrefix, + strlen(kEagerCustomCodePrefix)) == 0; } TfLiteIntArray* ConvertVectorToTfLiteIntArray(const std::vector<int>& input) { |