fix doc of ms_function
This commit is contained in:
parent
4be29b46b5
commit
743bbb4e18
|
@ -491,8 +491,8 @@ def ms_function(fn=None, input_signature=None, hash_args=None, jit_config=None):
|
||||||
... return fn(a)
|
... return fn(a)
|
||||||
... return inner_fn(x)
|
... return inner_fn(x)
|
||||||
...
|
...
|
||||||
... inputs = Tensor(np.ones([10, 10, 10]).astype(np.float32))
|
>>> inputs = Tensor(np.ones([10, 10, 10]).astype(np.float32))
|
||||||
... for i in range(10):
|
>>> for i in range(10):
|
||||||
... closure_fn(inputs, func)
|
... closure_fn(inputs, func)
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue