Add erf to flex model.

PiperOrigin-RevId: 252985034
2019-06-13 01:40:17 -07:00 · 2019-06-13 01:40:17 -07:00 · c02a9dab45
commit c02a9dab45
parent ad27775640
3 changed files with 3 additions and 0 deletions
--- a/tensorflow/contrib/makefile/tf_op_files.txt
+++ b/tensorflow/contrib/makefile/tf_op_files.txt
@ -61,6 +61,7 @@ tensorflow/core/kernels/cwise_op_cosh.cc
 tensorflow/core/kernels/cwise_op_div.cc
 tensorflow/core/kernels/cwise_op_equal_to_1.cc
 tensorflow/core/kernels/cwise_op_equal_to_2.cc
+tensorflow/core/kernels/cwise_op_erf.cc
 tensorflow/core/kernels/cwise_op_exp.cc
 tensorflow/core/kernels/cwise_op_floor.cc
 tensorflow/core/kernels/cwise_op_floor_div.cc
--- a/tensorflow/core/kernels/BUILD
+++ b/tensorflow/core/kernels/BUILD
@ -6126,6 +6126,7 @@ filegroup(
        "cwise_op_equal_to_2.cc",
        "cwise_op_not_equal_to_1.cc",
        "cwise_op_not_equal_to_2.cc",
+        "cwise_op_erf.cc",
        "cwise_op_exp.cc",
        "cwise_op_floor.cc",
        "cwise_op_floor_div.cc",
--- a/tensorflow/lite/delegates/flex/whitelisted_flex_ops.cc
+++ b/tensorflow/lite/delegates/flex/whitelisted_flex_ops.cc
@ -110,6 +110,7 @@ bool IsWhitelistedFlexOp(const std::string& tensorflow_op_name) {
          "EnsureShape",
          "Enter",
          "Equal",
+          "Erf",
          "Exit",
          "Exp",
          "ExpandDims",