Searched refs:ndarray (Results 1 – 3 of 3) sorted by relevance
| /llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/tools/ |
| H A D | mlir_pytaco_utils.py | 117 ) -> Tuple[int, int, np.ndarray, np.ndarray, np.ndarray]:
155 def coo_tensor_to_sparse_tensor(np_shape: np.ndarray, np_values: np.ndarray, argument
156 np_indices: np.ndarray, np_perm: np.ndarray, argument
157 np_sparse: np.ndarray) -> int: argument
271 type: str) -> Tuple[ctypes.c_void_p, np.ndarray]:
|
| H A D | mlir_pytaco.py | 175 def _ctype_pointer_from_array(array: np.ndarray) -> ctypes.pointer: argument
353 def get_permutation_and_sparsity(self) -> Tuple[np.ndarray, np.ndarray]:
1165 def to_array(self) -> np.ndarray:
1179 def from_array(array: np.ndarray) -> "Tensor": argument
1448 assert isinstance(result, np.ndarray)
|
| /llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/python/ |
| H A D | test_stress.py | 76 def get_RankedTensorType_of_nparray(self, nparray: np.ndarray) -> ir.RankedTensorType: argument
151 def run(self, np_arg0: np.ndarray) -> np.ndarray: argument
|