Fix docstrings in tvm.relay.cast_like (#8262)

* Fix docstrings in tvm.relay.cast_like * use correct formats to avoid chaos * Fix docstrings in tvm.relay.cast_like * use correct formats to avoid chaos
apache · Jun 16, 2021 · d05fdc5 · d05fdc5
1 parent a8663d2
commit d05fdc5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/tvm/relay/op/transform.py b/python/tvm/relay/op/transform.py
@@ -48,12 +48,15 @@ def cast(data, dtype):
 
 def cast_like(data, dtype_like):
     """Cast input tensor to data type of another tensor.
+
     Parameters
     ----------
     data : relay.Expr
         The input data to the operator.
+
     dtype_like: relay.Expr
         The tensor to cast to.
+
     Returns
     -------
     result : relay.Expr