| 700 | |
| 701 | |
| 702 | class Tensor(ABC): |
| 703 | ndarray: NDArray |
| 704 | data_type: DataType |
| 705 | |
| 706 | @abstractmethod |
| 707 | def astype(self, data_type: DataType) -> Self: ... |
| 708 | @abstractmethod |
| 709 | def permute(self, n_head: int, n_head_kv: int) -> Self: ... |
| 710 | @abstractmethod |
| 711 | def permute_part(self, n_part: int, n_head: int, n_head_kv: int) -> Self: ... |
| 712 | @abstractmethod |
| 713 | def part(self, n_part: int) -> Self: ... |
| 714 | @abstractmethod |
| 715 | def to_ggml(self) -> GGMLCompatibleTensor: ... |
| 716 | |
| 717 | |
| 718 | def bf16_to_fp32(bf16_arr: np.ndarray[Any, np.dtype[np.uint16]]) -> NDArray: |
nothing calls this directly
no outgoing calls
no test coverage detected