- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for CrossReplicaAndPartition (0.18 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
Arg<TF_Int32Tensor, [{Groups between which the reductions are performed.}]>:$group_assignment, TF_AnyStrAttrOf<["Min", "Max", "Mul", "Add", "Mean"]>:$reduce_op, TF_AnyStrAttrOf<["CrossReplica", "CrossReplicaAndPartition"]>:$mode ); let results = (outs TensorOf<[TF_Bfloat16, TF_Float16, TF_Float32, TF_Int32, TF_Uint32]>:$output ); TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)