MCPcopy Create free account
hub / github.com/ml-explore/mlx / unbatch_vjp_second

Function unbatch_vjp_second

benchmarks/python/batch_matmul_bench.py:46–47  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

44 time_fn(unbatch_vjp_first)
45
46 def unbatch_vjp_second():
47 return mx.matmul(mx.transpose(a), c)
48
49 time_fn(unbatch_vjp_second)
50

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected