(size: dict, dtype: torch.dtype, device: str, seed: int = 42)
| 209 | |
| 210 | |
| 211 | def gen_softmax_inputs(size: dict, dtype: torch.dtype, device: str, seed: int = 42) -> dict: |
| 212 | torch.manual_seed(seed) |
| 213 | rows, cols = size["rows"], size["cols"] |
| 214 | x = torch.randn(rows, cols, device=device, dtype=dtype) |
| 215 | return {"x": x} |
| 216 | |
| 217 | |
| 218 | def gen_layernorm_inputs(size: dict, dtype: torch.dtype, device: str, seed: int = 42) -> dict: |
nothing calls this directly
no outgoing calls
no test coverage detected