DefaultValuedOptionalAttr<BoolAttr, "false">:$keep_dims
  );

  let results = (outs
    Res<TF_BoolTensor, [{The reduced tensor.}]>:$output
  );

  TF_DerivedOperandTypeAttr Tidx = TF_DerivedOperandTypeAttr<1>;

  let hasVerifier = 1;
}

def TF_AllToAllOp : TF_Op<"AllToAll", [Pure, TF_NoConstantFold]> {
  let summary = "An Op to exchange data across TPU replicas.";