(inputs: dict)
| 290 | return reference.softmax_ref(inputs["x"]) |
| 291 | |
| 292 | def _ref_layernorm(inputs: dict) -> torch.Tensor: |
| 293 | import reference |
| 294 | return reference.layernorm_ref(inputs["x"], inputs["weight"], inputs["bias"]) |
| 295 | |
| 296 | def _ref_flash_attention(inputs: dict) -> torch.Tensor: |
| 297 | import reference |
nothing calls this directly
no outgoing calls
no test coverage detected