Formatting

Author: parsifal-47

include/tvm/relax/attrs/nn.h

@@ -552,8 +552,7 @@ struct AttentionAttrs : public tvm::AttrsNode<AttentionAttrs> {
   Optional<IntImm> window_size;
 
   TVM_DECLARE_ATTRS(AttentionAttrs, "relax.attrs.AttentionAttrs") {
-    TVM_ATTR_FIELD(bias).describe(
-        "The input bias tensor.");
+    TVM_ATTR_FIELD(bias).describe("The input bias tensor.");
     TVM_ATTR_FIELD(scale).describe(
         "The custom scale applied before the softmax. The default value is 1 / sqrt(head_dim).");
     TVM_ATTR_FIELD(causal_mask)

src/relax/op/nn/attention.cc

@@ -36,8 +36,7 @@ Expr attention(Expr query, Expr key, Expr value, Optional<Expr> bias, Optional<F
   attrs->window_size = window_size;
   attrs->bias = bias;
 
-  return Call(Op::Get("relax.nn.attention"),
-              {std::move(query), std::move(key), std::move(value)},
+  return Call(Op::Get("relax.nn.attention"), {std::move(query), std::move(key), std::move(value)},
               Attrs(attrs), {});
 }