Searched refs:complex64 (Results 1 – 8 of 8) sorted by relevance
| /llvm-project-15.0.7/mlir/test/python/ |
| H A D | execution_engine.py | 379 arg1 = np.array([5.+6.j]).astype(np.complex64)
380 arg2 = np.array([7.+8.j]).astype(np.complex64)
381 arg3 = np.array([0.+0.j]).astype(np.complex64)
401 np.dtype(np.complex64))
|
| /llvm-project-15.0.7/mlir/include/mlir/ExecutionEngine/ |
| H A D | SparseTensorUtils.h | 74 using complex64 = std::complex<double>; variable
101 DO(C64, complex64) \
|
| /llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/ |
| H A D | test_tensor_complex.py | 13 all_types = [pt.complex64, pt.complex128]
|
| /llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/tools/ |
| H A D | mlir_pytaco_api.py | 45 complex64 = mlir_pytaco.DType(mlir_pytaco.Type.COMPLEX64) variable
|
| H A D | mlir_pytaco_utils.py | 97 (np.complex64, c_lib.convertToMLIRSparseTensorC32,
|
| H A D | mlir_pytaco.py | 78 COMPLEX64 = np.complex64
91 np.float32, np.float64, np.complex64, np.complex128]
154 np.complex64: Type.COMPLEX64,
|
| /llvm-project-15.0.7/mlir/python/mlir/runtime/ |
| H A D | np_to_memref.py | 30 if dtp is np.dtype(np.complex64):
|
| /llvm-project-15.0.7/mlir/lib/ExecutionEngine/ |
| H A D | SparseTensorUtils.cpp | 1672 CASE_SECSAME(OverheadType::kU64, PrimaryType::kC64, uint64_t, complex64); in _mlir_ciface_newSparseTensor()
|