Home
last modified time | relevance | path

Searched defs:ndarray (Results 1 – 3 of 3) sorted by relevance

/llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/tools/
H A Dmlir_pytaco_utils.py155 def coo_tensor_to_sparse_tensor(np_shape: np.ndarray, np_values: np.ndarray,
156 np_indices: np.ndarray, np_perm: np.ndarray,
157 np_sparse: np.ndarray) -> int:
H A Dmlir_pytaco.py175 def _ctype_pointer_from_array(array: np.ndarray) -> ctypes.pointer:
1179 def from_array(array: np.ndarray) -> "Tensor":
/llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/python/
H A Dtest_stress.py76 def get_RankedTensorType_of_nparray(self, nparray: np.ndarray) -> ir.RankedTensorType:
151 def run(self, np_arg0: np.ndarray) -> np.ndarray: