[KERNEL_GEN][CPU] Add `platform` placeholder to op_definitions/*.mlir.tmpl.
PiperOrigin-RevId: 355130370
Change-Id: I4ae4bfeaeef702e66c0064f8d594acc718d4fc0d
diff --git a/tensorflow/core/kernels/mlir_generated/build_defs.bzl b/tensorflow/core/kernels/mlir_generated/build_defs.bzl
index 3f6bb5d..851e0b6 100644
--- a/tensorflow/core/kernels/mlir_generated/build_defs.bzl
+++ b/tensorflow/core/kernels/mlir_generated/build_defs.bzl
@@ -49,9 +49,11 @@
inputs = [ctx.file.template],
outputs = [ctx.outputs.out],
command = (
- (("cat %s | sed 's/_elem_type/_%s/g' | sed 's/elem_type/%s/g' | " +
- "sed 's/_output_type/_%s/g' | sed 's/output_type/%s/g' > %s")) % (
+ (("cat %s | sed 's/platform/%s/g' | sed 's/_elem_type/_%s/g' | " +
+ "sed 's/elem_type/%s/g' | " + "sed 's/_output_type/_%s/g' | " +
+ "sed 's/output_type/%s/g' > %s")) % (
ctx.file.template.path,
+ ctx.attr.platform.upper(),
ctx.attr.type,
mlir_type,
ctx.attr.output_type,
@@ -68,22 +70,26 @@
"template": attr.label(mandatory = True, allow_single_file = True),
"type": attr.string(mandatory = True),
"output_type": attr.string(mandatory = True),
+ "platform": attr.string(mandatory = True),
"out": attr.output(mandatory = True),
},
)
-def _gen_mlir_op(name, type, output_type):
+def _gen_mlir_op(name, type, platform, output_type):
_gen_mlir_op_rule(
- name = "generate_{name}_{type}_{output_type}_mlir".format(
+ name = "generate_{name}_{platform}_{type}_{output_type}_mlir".format(
name = name,
+ platform = platform,
type = type,
output_type = output_type,
),
template = "op_definitions/{name}.mlir.tmpl".format(name = name),
+ platform = platform,
type = type,
output_type = output_type,
- out = "{name}_{type}_{output_type}.mlir".format(
+ out = "{name}_{platform}_{type}_{output_type}.mlir".format(
name = name,
+ platform = platform,
type = type,
output_type = output_type,
),
@@ -177,6 +183,7 @@
tile_size,
output_types = None,
tags = [],
+ platform = "gpu",
unroll_factors = None,
extra_args = []):
""" Generate a library with kernels for a specific tensorflow op.
@@ -190,6 +197,7 @@
entry in output_types. By default, output_types = types is
assumed.
tags: The tags which should be added to the library.
+ platform: Platform for which to compile, i.e. "cpu" or "gpu"
unroll_factors: The unrolling specification, e.g. "4,4"
extra_args: Extra arguments to pass to the generator tool.
"""
@@ -200,17 +208,20 @@
for (type, output_type) in zip(types, output_types):
_gen_mlir_op(
name = name,
+ platform = platform,
type = type,
output_type = output_type,
)
_gen_kernel_fatbin_rule(
- name = "{name}_{type}_{output_type}_kernel_generator".format(
+ name = "{name}_{platform}_{type}_{output_type}_kernel_generator".format(
name = name,
+ platform = platform,
type = type,
output_type = output_type,
),
- mlir_op = "{name}_{type}_{output_type}.mlir".format(
+ mlir_op = "{name}_{platform}_{type}_{output_type}.mlir".format(
name = name,
+ platform = platform,
type = type,
output_type = output_type,
),
@@ -223,8 +234,9 @@
# We have to use a sh_test instead of build_test because it doesn't properly find the dependent targets.
native.sh_test(
- name = "{name}_{type}_{output_type}_gen_test".format(
+ name = "{name}_{platform}_{type}_{output_type}_gen_test".format(
name = name,
+ platform = platform,
type = type,
output_type = output_type,
),
@@ -232,16 +244,18 @@
tags = ["no_rocm"],
args = [
"$(location //tensorflow/compiler/mlir/tools/kernel_gen:tf_to_kernel)",
- "$(location {name}_{type}_{output_type}.mlir)".format(
+ "$(location {name}_{platform}_{type}_{output_type}.mlir)".format(
name = name,
+ platform = platform,
type = type,
output_type = output_type,
),
],
size = "medium",
data = [
- ":{name}_{type}_{output_type}.mlir".format(
+ ":{name}_{platform}_{type}_{output_type}.mlir".format(
name = name,
+ platform = platform,
type = type,
output_type = output_type,
),
@@ -252,8 +266,9 @@
native.cc_library(
name = name + "_kernels",
compatible_with = get_compatible_with_cloud(),
- deps = if_gpu_is_configured([":{name}_{type}_{output_type}_kernel_generator".format(
+ deps = if_gpu_is_configured([":{name}_{platform}_{type}_{output_type}_kernel_generator".format(
name = name,
+ platform = platform,
type = type,
output_type = output_type,
) for (type, output_type) in zip(types, output_types)]),
diff --git a/tensorflow/core/kernels/mlir_generated/gpu_ops_base.h b/tensorflow/core/kernels/mlir_generated/gpu_ops_base.h
index 47bcb65..3147a79 100644
--- a/tensorflow/core/kernels/mlir_generated/gpu_ops_base.h
+++ b/tensorflow/core/kernels/mlir_generated/gpu_ops_base.h
@@ -143,7 +143,7 @@
};
#define MLIR_FUNCTION(tf_op, mlir_type, mlir_output_type) \
- _mlir_ciface_##tf_op##_##mlir_type##_##mlir_output_type
+ _mlir_ciface_##tf_op##_GPU_##mlir_type##_##mlir_output_type
#define REGISTER_ALIASED_KERNEL(tf_op, mlir_op, mlir_type, mlir_output_type, \
data_type) \
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/abs.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/abs.mlir.tmpl
index b25d538..a6549d4 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/abs.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/abs.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Abs_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Abs_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Abs"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/acos.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/acos.mlir.tmpl
index c6c79c1..5cf4773 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/acos.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/acos.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Acos_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Acos_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Acos"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/acosh.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/acosh.mlir.tmpl
index 5c4c528..618a662 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/acosh.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/acosh.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Acosh_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Acosh_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Acosh"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/add_v2.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/add_v2.mlir.tmpl
index 70a193d..85fff1a 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/add_v2.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/add_v2.mlir.tmpl
@@ -1,4 +1,4 @@
-func @AddV2_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @AddV2_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.AddV2"(%arg0, %arg1) {T = elem_type, device = ""}
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/angle.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/angle.mlir.tmpl
index 6f8e4a3..8c53421 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/angle.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/angle.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Angle_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Angle_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Angle"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/asin.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/asin.mlir.tmpl
index 6c18ef9..042beab 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/asin.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/asin.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Asin_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Asin_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Asin"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/asinh.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/asinh.mlir.tmpl
index c1f5ed5..cb4ffe4 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/asinh.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/asinh.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Asinh_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Asinh_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Asinh"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/atan.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/atan.mlir.tmpl
index 3315426..9cbf018 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/atan.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/atan.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Atan_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Atan_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Atan"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/atan2.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/atan2.mlir.tmpl
index dc549ff..b611971 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/atan2.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/atan2.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Atan2_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @Atan2_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Atan2"(%arg0, %arg1)
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/atanh.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/atanh.mlir.tmpl
index 66661ad..a634f11 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/atanh.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/atanh.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Atanh_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Atanh_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Atanh"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/bias_add.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/bias_add.mlir.tmpl
index aa34480..4d194ab 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/bias_add.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/bias_add.mlir.tmpl
@@ -1,4 +1,4 @@
-func @BiasAdd_elem_type_output_type(%arg0: tensor<*x*xelem_type>, %arg1: tensor<*xelem_type>)
+func @BiasAdd_platform_elem_type_output_type(%arg0: tensor<*x*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*x*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.BiasAdd"(%arg0, %arg1)
: (tensor<*x*xelem_type>, tensor<*xelem_type>) -> tensor<*x*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/bitwise_and.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/bitwise_and.mlir.tmpl
index 4066cc2..3d25565 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/bitwise_and.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/bitwise_and.mlir.tmpl
@@ -1,4 +1,4 @@
-func @BitwiseAnd_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @BitwiseAnd_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.BitwiseAnd"(%arg0, %arg1)
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/bitwise_or.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/bitwise_or.mlir.tmpl
index 6153916..e1dcf79 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/bitwise_or.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/bitwise_or.mlir.tmpl
@@ -1,4 +1,4 @@
-func @BitwiseOr_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @BitwiseOr_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.BitwiseOr"(%arg0, %arg1)
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/bitwise_xor.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/bitwise_xor.mlir.tmpl
index 20fbd82..32382ab 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/bitwise_xor.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/bitwise_xor.mlir.tmpl
@@ -1,4 +1,4 @@
-func @BitwiseXor_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @BitwiseXor_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.BitwiseXor"(%arg0, %arg1)
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/cast.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/cast.mlir.tmpl
index 24092ace..018798e 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/cast.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/cast.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Cast_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Cast_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Cast"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/ceil.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/ceil.mlir.tmpl
index e15a1e9..56d0407 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/ceil.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/ceil.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Ceil_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Ceil_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Ceil"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/complex.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/complex.mlir.tmpl
index e340251..86fe83f 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/complex.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/complex.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Complex_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @Complex_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Complex"(%arg0, %arg1) {}
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/complex_abs.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/complex_abs.mlir.tmpl
index 8db9686..8fe10ab 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/complex_abs.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/complex_abs.mlir.tmpl
@@ -1,4 +1,4 @@
-func @ComplexAbs_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @ComplexAbs_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.ComplexAbs"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/conj.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/conj.mlir.tmpl
index 9da64eb..b654909 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/conj.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/conj.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Conj_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Conj_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Conj"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/cos.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/cos.mlir.tmpl
index 14833ee..f0f8783 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/cos.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/cos.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Cos_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Cos_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Cos"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/cosh.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/cosh.mlir.tmpl
index aa34dd2..582b1e9 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/cosh.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/cosh.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Cosh_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Cosh_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Cosh"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/digamma.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/digamma.mlir.tmpl
index 61a2498..4218401 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/digamma.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/digamma.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Digamma_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Digamma_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Digamma"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/div.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/div.mlir.tmpl
index 14cd5cc..9f0565c 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/div.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/div.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Div_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @Div_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Div"(%arg0, %arg1) {T = elem_type, device = ""}
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/equal.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/equal.mlir.tmpl
index d841812..2b653a9 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/equal.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/equal.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Equal_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @Equal_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Equal"(%arg0, %arg1) {T = elem_type, device = ""}
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/erf.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/erf.mlir.tmpl
index 8d1a8ad..df00973 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/erf.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/erf.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Erf_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Erf_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Erf"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/erfc.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/erfc.mlir.tmpl
index d531669..305c6a3 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/erfc.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/erfc.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Erfc_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Erfc_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Erfc"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/exp.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/exp.mlir.tmpl
index d5979c1..046e94f 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/exp.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/exp.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Exp_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Exp_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Exp"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/expm1.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/expm1.mlir.tmpl
index 389d399..05d1a76 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/expm1.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/expm1.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Expm1_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Expm1_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Expm1"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/floor.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/floor.mlir.tmpl
index f673b38..40bf1fc 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/floor.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/floor.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Floor_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Floor_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Floor"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/floor_div.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/floor_div.mlir.tmpl
index d432c33..1bb2b0e 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/floor_div.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/floor_div.mlir.tmpl
@@ -1,4 +1,4 @@
-func @FloorDiv_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @FloorDiv_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.FloorDiv"(%arg0, %arg1) {T = elem_type, device = ""}
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/greater.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/greater.mlir.tmpl
index 5632d1e..f5b9172 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/greater.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/greater.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Greater_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @Greater_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Greater"(%arg0, %arg1) {T = elem_type, device = ""}
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/greater_equal.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/greater_equal.mlir.tmpl
index a9edf46..d83775e 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/greater_equal.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/greater_equal.mlir.tmpl
@@ -1,4 +1,4 @@
-func @GreaterEqual_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @GreaterEqual_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.GreaterEqual"(%arg0, %arg1) {T = elem_type, device = ""}
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/imag.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/imag.mlir.tmpl
index 7ccffa6..a22e7fe 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/imag.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/imag.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Imag_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Imag_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Imag"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/invert.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/invert.mlir.tmpl
index f68e842..b64f99e 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/invert.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/invert.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Invert_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Invert_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Invert"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/is_finite.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/is_finite.mlir.tmpl
index 9d896c0..11b4553 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/is_finite.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/is_finite.mlir.tmpl
@@ -1,4 +1,4 @@
-func @IsFinite_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @IsFinite_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.IsFinite"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/is_inf.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/is_inf.mlir.tmpl
index 7046082..af86b39 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/is_inf.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/is_inf.mlir.tmpl
@@ -1,4 +1,4 @@
-func @IsInf_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @IsInf_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.IsInf"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/is_nan.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/is_nan.mlir.tmpl
index 4e7592f..4bae71d 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/is_nan.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/is_nan.mlir.tmpl
@@ -1,4 +1,4 @@
-func @IsNan_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @IsNan_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.IsNan"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/left_shift.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/left_shift.mlir.tmpl
index 177e2dd..3c2b37c 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/left_shift.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/left_shift.mlir.tmpl
@@ -1,4 +1,4 @@
-func @LeftShift_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @LeftShift_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.LeftShift"(%arg0, %arg1) {T = elem_type, device = ""}
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/less.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/less.mlir.tmpl
index 7a8db8d..349ffe7 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/less.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/less.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Less_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @Less_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Less"(%arg0, %arg1) {T = elem_type, device = ""}
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/less_equal.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/less_equal.mlir.tmpl
index e4b4bb6..d23dd26 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/less_equal.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/less_equal.mlir.tmpl
@@ -1,4 +1,4 @@
-func @LessEqual_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @LessEqual_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.LessEqual"(%arg0, %arg1) {T = elem_type, device = ""}
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/lgamma.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/lgamma.mlir.tmpl
index aee4e34..3e7e721 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/lgamma.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/lgamma.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Lgamma_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Lgamma_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Lgamma"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/log.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/log.mlir.tmpl
index 3124e36..f5743d7 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/log.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/log.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Log_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Log_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Log"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/log1p.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/log1p.mlir.tmpl
index 189e1ad..ffbf7e5 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/log1p.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/log1p.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Log1p_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Log1p_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Log1p"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/logical_and.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/logical_and.mlir.tmpl
index 67aad3c..d034ae1 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/logical_and.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/logical_and.mlir.tmpl
@@ -1,4 +1,4 @@
-func @LogicalAnd_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @LogicalAnd_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.LogicalAnd"(%arg0, %arg1)
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/logical_not.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/logical_not.mlir.tmpl
index 90714f9..9428ffa 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/logical_not.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/logical_not.mlir.tmpl
@@ -1,4 +1,4 @@
-func @LogicalNot_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @LogicalNot_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.LogicalNot"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/logical_or.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/logical_or.mlir.tmpl
index 17307a9..9d54564 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/logical_or.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/logical_or.mlir.tmpl
@@ -1,4 +1,4 @@
-func @LogicalOr_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @LogicalOr_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.LogicalOr"(%arg0, %arg1)
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/maximum.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/maximum.mlir.tmpl
index f43ced1..fbff94d 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/maximum.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/maximum.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Maximum_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @Maximum_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Maximum"(%arg0, %arg1) {T = elem_type, device = ""}
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/minimum.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/minimum.mlir.tmpl
index ab30c00..d2fc8e1 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/minimum.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/minimum.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Minimum_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @Minimum_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Minimum"(%arg0, %arg1) {T = elem_type, device = ""}
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/mul.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/mul.mlir.tmpl
index caaf13e..36d4514 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/mul.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/mul.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Mul_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @Mul_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Mul"(%arg0, %arg1) {T = elem_type, device = ""}
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/neg.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/neg.mlir.tmpl
index 3147e3c..7bc21e1 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/neg.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/neg.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Neg_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Neg_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Neg"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/not_equal.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/not_equal.mlir.tmpl
index ee05d7b..e3e385b 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/not_equal.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/not_equal.mlir.tmpl
@@ -1,4 +1,4 @@
-func @NotEqual_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @NotEqual_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.NotEqual"(%arg0, %arg1) {T = elem_type, device = ""}
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/pow.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/pow.mlir.tmpl
index f09d8a4..a518d84 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/pow.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/pow.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Pow_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @Pow_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Pow"(%arg0, %arg1) {T = elem_type, device = ""}
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/real.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/real.mlir.tmpl
index 7363e1a..6865733 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/real.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/real.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Real_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Real_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Real"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/reciprocal.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/reciprocal.mlir.tmpl
index a366f08..7c6850c 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/reciprocal.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/reciprocal.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Reciprocal_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Reciprocal_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Reciprocal"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/relu.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/relu.mlir.tmpl
index a5c0423..79a6f96 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/relu.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/relu.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Relu_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Relu_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Relu"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/right_shift.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/right_shift.mlir.tmpl
index 65c1d8b..c17b26d 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/right_shift.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/right_shift.mlir.tmpl
@@ -1,4 +1,4 @@
-func @RightShift_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @RightShift_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.RightShift"(%arg0, %arg1) {T = elem_type, device = ""}
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/rint.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/rint.mlir.tmpl
index fc574e8..bacf6d76 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/rint.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/rint.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Rint_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Rint_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Rint"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/round.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/round.mlir.tmpl
index a4832df..a9e07df 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/round.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/round.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Round_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Round_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Round"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/rsqrt.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/rsqrt.mlir.tmpl
index fbd80ce..a360e25 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/rsqrt.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/rsqrt.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Rsqrt_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Rsqrt_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Rsqrt"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/sigmoid.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/sigmoid.mlir.tmpl
index 66da269..86c5415 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/sigmoid.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/sigmoid.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Sigmoid_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Sigmoid_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Sigmoid"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/sign.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/sign.mlir.tmpl
index a5cad51..9e29cec 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/sign.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/sign.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Sign_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Sign_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Sign"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/sin.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/sin.mlir.tmpl
index 4bb4220..f830afb 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/sin.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/sin.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Sin_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Sin_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Sin"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/sinh.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/sinh.mlir.tmpl
index fc3c275..ebef8e6 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/sinh.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/sinh.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Sinh_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Sinh_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Sinh"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/sqrt.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/sqrt.mlir.tmpl
index 4f3049b..0f735c4 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/sqrt.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/sqrt.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Sqrt_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Sqrt_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Sqrt"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/square.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/square.mlir.tmpl
index 1781366..4632166 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/square.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/square.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Square_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Square_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Square"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/squared_difference.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/squared_difference.mlir.tmpl
index 1475bc7..50a50b5 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/squared_difference.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/squared_difference.mlir.tmpl
@@ -1,4 +1,4 @@
-func @SquaredDifference_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @SquaredDifference_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.SquaredDifference"(%arg0, %arg1) {T = elem_type, device = ""}
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/sub.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/sub.mlir.tmpl
index 866cf32..dbbc238 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/sub.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/sub.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Sub_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
+func @Sub_platform_elem_type_output_type(%arg0: tensor<*xelem_type>, %arg1: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Sub"(%arg0, %arg1) {T = elem_type, device = ""}
: (tensor<*xelem_type>, tensor<*xelem_type>) -> tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/tan.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/tan.mlir.tmpl
index f48e058..776687d 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/tan.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/tan.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Tan_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Tan_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Tan"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
diff --git a/tensorflow/core/kernels/mlir_generated/op_definitions/tanh.mlir.tmpl b/tensorflow/core/kernels/mlir_generated/op_definitions/tanh.mlir.tmpl
index 10cb9b7..1c4e87e 100644
--- a/tensorflow/core/kernels/mlir_generated/op_definitions/tanh.mlir.tmpl
+++ b/tensorflow/core/kernels/mlir_generated/op_definitions/tanh.mlir.tmpl
@@ -1,4 +1,4 @@
-func @Tanh_elem_type_output_type(%arg0: tensor<*xelem_type>)
+func @Tanh_platform_elem_type_output_type(%arg0: tensor<*xelem_type>)
-> tensor<*xoutput_type> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.Tanh"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
return %0 : tensor<*xoutput_type>
