commit 31583920dcdeb54ffdb34acb1ce6b1db546ad33c
author Ken Franko <kfranko@google.com> Fri May 15 13:34:26 2020 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Fri May 15 13:39:32 2020 -0700
tree 8622c4e5fce7f316d0164257ad8f44cfc4b27124
parent dc1c299833317401e14a1651ed906164464425c8

Create _HostComputeMlir for use in TF MLIR.

PiperOrigin-RevId: 311792286
Change-Id: I6ec57f9b23c17dd52e756ead4ddfad58ecdb2f76

tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

diff --git a/tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td b/tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
index aa1601c..82282bb 100644
--- a/tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
+++ b/tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
@@ -10586,6 +10586,27 @@
   TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>;
 }
 
+def TF__HostComputeMlirOp : TF_Op<"_HostComputeMlir", []> {
+  let summary = "A host-side computation called from a TPU device.";
+
+  let description = [{
+  }];
+
+  let arguments = (ins
+    Variadic<TF_Tensor>:$inputs,
+
+    StrAttr:$key,
+    DefaultValuedAttr<I64Attr, "0">:$tpu_core
+  );
+
+  let results = (outs
+    Variadic<TF_Tensor>:$outputs
+  );
+
+  TF_DerivedOperandTypeListAttr Tinputs = TF_DerivedOperandTypeListAttr<0>;
+  TF_DerivedResultTypeListAttr Toutputs = TF_DerivedResultTypeListAttr<0>;
+}
+
 def TF__RecvTPUEmbeddingActivationsOp : TF_Op<"_RecvTPUEmbeddingActivations", []> {
   let summary = "An op that receives embeddng activations on the TPU.";