Fix docstrings in tvm.relay.cast_like (apache#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
neo-ai · Jun 17, 2021 · b72b71c · b72b71c
1 parent 6d631ac
commit b72b71c
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