diff --git a/tensorflow/contrib/makefile/tf_op_files.txt b/tensorflow/contrib/makefile/tf_op_files.txt index ac54c0c3a80..88dbff483f2 100644 --- 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 diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index 6a51a611948..644e591100c 100644 --- 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", diff --git a/tensorflow/lite/delegates/flex/whitelisted_flex_ops.cc b/tensorflow/lite/delegates/flex/whitelisted_flex_ops.cc index 30cf79b4a05..2d73b7ee3a3 100644 --- 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",