Home
last modified time | relevance | path

Searched refs:complex64 (Results 1 – 6 of 6) sorted by relevance

/llvm-project/mlir/lib/ExecutionEngine/
H A DRunnerUtils.cpp69 _mlir_ciface_printMemrefShapeC64(UnrankedMemRefType<impl::complex64> *M) { in _mlir_ciface_printMemrefShapeC64()
71 printMemRefMetaData(std::cout, DynamicMemRefType<impl::complex64>(*M)); in _mlir_ciface_printMemrefShapeC64()
123 _mlir_ciface_printMemrefC64(UnrankedMemRefType<impl::complex64> *M) { in _mlir_ciface_printMemrefC64()
169 UnrankedMemRefType<impl::complex64> descriptor = {rank, ptr}; in printMemrefC64()
218 _mlir_ciface_printMemref1dC64(StridedMemRefType<impl::complex64, 1> *M) { in _mlir_ciface_printMemref1dC64() argument
283 _mlir_ciface_verifyMemRefC64(UnrankedMemRefType<impl::complex64> *actual, in _mlir_ciface_verifyMemRefC64()
284 UnrankedMemRefType<impl::complex64> *expected) { in _mlir_ciface_verifyMemRefC64()
325 UnrankedMemRefType<impl::complex64> actualDesc = {rank, actualPtr}; in verifyMemRefC64()
326 UnrankedMemRefType<impl::complex64> expectedDesc = {rank, expectedPtr}; in verifyMemRefC64()
H A DSparseTensorRuntime.cpp281 CASE_SECSAME(OverheadType::kU64, PrimaryType::kC64, uint64_t, complex64); in _mlir_ciface_newSparseTensor()
/llvm-project/mlir/include/mlir/ExecutionEngine/
H A DRunnerUtils.h82 using complex64 = std::complex<double>; variable
368 _mlir_ciface_printMemrefShapeC64(UnrankedMemRefType<impl::complex64> *m);
391 _mlir_ciface_printMemrefC64(UnrankedMemRefType<impl::complex64> *m);
427 _mlir_ciface_printMemref1dC64(StridedMemRefType<impl::complex64, 1> *m);
455 _mlir_ciface_verifyMemRefC64(UnrankedMemRefType<impl::complex64> *actual,
456 UnrankedMemRefType<impl::complex64> *expected);
/llvm-project/mlir/test/python/
H A Dexecution_engine.py508 arg1 = np.array([5.0 + 6.0j]).astype(np.complex64)
509 arg2 = np.array([7.0 + 8.0j]).astype(np.complex64)
510 arg3 = np.array([0.0 + 0.0j]).astype(np.complex64)
531 npout = unranked_memref_to_numpy(arg3_memref_ptr[0], np.dtype(np.complex64))
/llvm-project/mlir/include/mlir/Dialect/SparseTensor/IR/
H A DEnums.h78 using complex64 = std::complex<double>; variable
105 DO(C64, complex64) \
118 DO(C64, complex64, __VA_ARGS__) \
/llvm-project/mlir/python/mlir/runtime/
H A Dnp_to_memref.py51 if dtp == np.dtype(np.complex64):
67 return array.view("complex64")