experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update TF MLIR ODS for tf.CollectiveBcastRecv, tf.CollectiveBcastSend, tf.CollectiveGather, tf.CollectiveReduce, and tf.TPUReplicatedInput. 

Ops have additional attributes with default values added to them in TensorFlow op registry.

PiperOrigin-RevId: 315802211
Change-Id: I20e404a40163ce4e596220fe5a76e1dfb74825c1
This commit is contained in:
Andy Ly 2020-06-10 17:31:58 -07:00 committed by TensorFlower Gardener
parent c3da1a69c3
commit 5d1368571f
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
View File

@ -1353,7 +1353,8 @@ def TF_CollectiveBcastRecvOp : TF_Op<"CollectiveBcastRecv", []> {
I64Attr:$group_key,
I64Attr:$instance_key,
TF_ShapeAttr:$shape,
DefaultValuedAttr<StrAttr, "auto">:$communication_hint
DefaultValuedAttr<StrAttr, "auto">:$communication_hint,
DefaultValuedAttr<F32Attr, "0.0f">:$timeout_seconds
);
let results = (outs
@ -1376,7 +1377,8 @@ def TF_CollectiveBcastSendOp : TF_Op<"CollectiveBcastSend", []> {
I64Attr:$group_key,
I64Attr:$instance_key,
TF_ShapeAttr:$shape,
DefaultValuedAttr<StrAttr, "auto">:$communication_hint
DefaultValuedAttr<StrAttr, "auto">:$communication_hint,
DefaultValuedAttr<F32Attr, "0.0f">:$timeout_seconds
);
let results = (outs
@ -1401,7 +1403,8 @@ Mutually accumulates multiple tensors of identical type and shape.
I64Attr:$group_key,
I64Attr:$instance_key,
TF_ShapeAttr:$shape,
DefaultValuedAttr<StrAttr, "auto">:$communication_hint
DefaultValuedAttr<StrAttr, "auto">:$communication_hint,
DefaultValuedAttr<F32Attr, "0.0f">:$timeout_seconds
);
let results = (outs
@ -1429,7 +1432,8 @@ Mutually reduces multiple tensors of identical type and shape.
TF_AnyStrAttrOf<["Id", "Div"]>:$final_op,
I64ArrayAttr:$subdiv_offsets,
DefaultValuedAttr<I64ArrayAttr, "{}">:$wait_for,
DefaultValuedAttr<StrAttr, "auto">:$communication_hint
DefaultValuedAttr<StrAttr, "auto">:$communication_hint,
DefaultValuedAttr<F32Attr, "0.0f">:$timeout_seconds
);
let results = (outs
@ -8977,7 +8981,8 @@ The above computation has a replicated input of two replicas.
Variadic<TF_Tensor>:$inputs,
DefaultValuedAttr<BoolAttr, "false">:$is_mirrored_variable,
DefaultValuedAttr<I64Attr, "-1">:$index
DefaultValuedAttr<I64Attr, "-1">:$index,
DefaultValuedAttr<BoolAttr, "false">:$is_packed
);
let results = (outs