- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for group_key (0.34 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
); let results = (outs TF_Int32Tensor:$group_size, TF_Int32Tensor:$group_key ); } def TF_CollectiveBcastRecvOp : TF_Op<"CollectiveBcastRecv", []> { let summary = "Receives a tensor value broadcast from another device."; let arguments = (ins I64Attr:$group_size, I64Attr:$group_key, I64Attr:$instance_key, TF_ShapeAttr:$shape,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)