Searched defs:input_type (Results 1 – 3 of 3) sorted by relevance
| /llvm-project/mlir/test/Integration/GPU/CUDA/sm90/python/ |
| H A D | matmul.py | 90 input_type=np.float16, argument
156 input_type=np.float16, argument
|
| /llvm-project/mlir/test/Integration/GPU/CUDA/sm90/python/tools/ |
| H A D | matmulBuilder.py | 132 input_type=np.float16, argument
164 input_type=np.float16, argument
664 generate_matmul_multistage( input_type=np.float16, output_type=np.float32, M=4096, N=4096, K=4096, BLOCK_M=128, BLOCK_N=128, BLOCK_K=64, num_stages=3, ) global() argument
[all...] |
| /llvm-project/pstl/test/std/numerics/numeric.ops/ |
| H A D | scan.pass.cpp | 80 typedef typename std::iterator_traits<InputIterator>::value_type input_type; in inclusive_scan_serial() typedef
|