Automated Code Change

PiperOrigin-RevId: 629336935
tensorflow · Apr 30, 2024 · 556f747 · 556f747
1 parent 8d393e7
commit 556f747
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/tensorflow/examples/custom_ops_doc/multiplex_1/multiplex_1_kernel.cc b/tensorflow/examples/custom_ops_doc/multiplex_1/multiplex_1_kernel.cc
@@ -14,7 +14,10 @@ limitations under the License.
 ==============================================================================*/
 
 #include "tensorflow/core/framework/op_kernel.h"
+#include "tensorflow/core/framework/op_requires.h"
 #include "tensorflow/core/framework/register_types.h"
+#include "tensorflow/core/framework/tensor.h"
+#include "tensorflow/core/framework/types.h"
 
 // Please use the appropriate namespace for your project
 namespace tensorflow {

diff --git a/tensorflow/examples/custom_ops_doc/multiplex_1/multiplex_1_op.cc b/tensorflow/examples/custom_ops_doc/multiplex_1/multiplex_1_op.cc
@@ -15,6 +15,8 @@ limitations under the License.
 
 #include "tensorflow/core/framework/op.h"
 #include "tensorflow/core/framework/shape_inference.h"
+#include "tensorflow/core/platform/status.h"
+#include "tsl/platform/errors.h"
 
 // Use a namespace when registering by prepending the
 // package's name to the op’s name and separate with a '>'.