MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / fwd

Function fwd

imperative/python/test/unit/xla/functional/test_xla_indexing.py:48–49  ·  view source on GitHub ↗
(inp)

Source from the content-addressed store, hash-verified

46
47 @jit.partial_trace(without_host=True, capture_as_const=True, backend="xla")
48 def fwd(inp):
49 return inp[index]
50
51 def func(inp, dout):
52 gm.attach([inp])

Callers 1

funcFunction · 0.70

Calls

no outgoing calls

Tested by

no test coverage detected