(inputs: dict)
| 306 | return reference.cross_entropy_ref(inputs["logits"], inputs["targets"]) |
| 307 | |
| 308 | def _ref_rotary_embedding(inputs: dict) -> torch.Tensor: |
| 309 | import reference |
| 310 | return reference.rotary_embedding_ref(inputs["x"], inputs["cos"], inputs["sin"]) |
| 311 | |
| 312 | def _ref_rmsnorm(inputs: dict) -> torch.Tensor: |
| 313 | import reference |
nothing calls this directly
no outgoing calls
no test coverage detected