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 inner_fn(x)
|
||||
...
|
||||
... inputs = Tensor(np.ones([10, 10, 10]).astype(np.float32))
|
||||
... for i in range(10):
|
||||
>>> inputs = Tensor(np.ones([10, 10, 10]).astype(np.float32))
|
||||
>>> for i in range(10):
|
||||
... closure_fn(inputs, func)
|
||||
"""
|
||||
|
||||
|
|
Loading…
Reference in New Issue