MCPcopy Create free account
hub / github.com/RightNow-AI/autokernel / _ref_softmax

Function _ref_softmax

bench.py:288–290  ·  view source on GitHub ↗
(inputs: dict)

Source from the content-addressed store, hash-verified

286 return reference.matmul_ref(inputs["A"], inputs["B"])
287
288def _ref_softmax(inputs: dict) -> torch.Tensor:
289 import reference
290 return reference.softmax_ref(inputs["x"])
291
292def _ref_layernorm(inputs: dict) -> torch.Tensor:
293 import reference

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected