@@ -16,16 +16,14 @@ def source_to_source(
16
16
as well e.g. (source="torch_frontend", target="ivy") or (source="torch_frontend", target="tensorflow") etc.
17
17
18
18
Args:
19
- ----
20
19
object: The object (class/function) to be translated.
21
20
source (str, optional): The source framework. Defaults to 'torch'.
22
21
target (str, optional): The target framework. Defaults to 'torch_frontend'.
23
22
profiling: Whether to add performance profiling.
24
23
25
24
Returns:
26
- -------
27
- The translated object.
28
- """
25
+ The translated object."""
26
+
29
27
from ._compiler import source_to_source as _source_to_source
30
28
31
29
return _source_to_source (
@@ -56,8 +54,7 @@ def trace_graph(
56
54
params_v = None ,
57
55
v = None
58
56
):
59
- """Takes `fn` and traces it into a more efficient composition of backend
60
- operations.
57
+ """Takes `fn` and traces it into a more efficient composition of backend operations.
61
58
62
59
Parameters
63
60
----------
@@ -127,8 +124,8 @@ def trace_graph(
127
124
>>> start = time.time()
128
125
>>> graph(x)
129
126
>>> print(time.time() - start)
130
- 0.0001785755157470703
131
- """
127
+ 0.0001785755157470703"""
128
+
132
129
from ._compiler import trace_graph as _trace_graph
133
130
134
131
return _trace_graph (
@@ -192,8 +189,8 @@ def transpile(
192
189
193
190
Returns
194
191
-------
195
- Either a transpiled Graph or a non-initialized LazyGraph.
196
- """
192
+ Either a transpiled Graph or a non-initialized LazyGraph."""
193
+
197
194
from ._compiler import transpile as _transpile
198
195
199
196
return _transpile (
0 commit comments