Searched refs:to_mkl_complex (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/aten/src/ATen/mkl/ |
| D | SparseBlas.cpp | 13 MKL_Complex to_mkl_complex(c10::complex<scalar_t> scalar) { in to_mkl_complex() function
133 to_mkl_complex<float, MKL_Complex8>(alpha), in mv()
137 to_mkl_complex<float, MKL_Complex8>(beta), in mv()
144 to_mkl_complex<double, MKL_Complex16>(alpha), in mv()
148 to_mkl_complex<double, MKL_Complex16>(beta), in mv()
163 operation, A, to_mkl_complex<float, MKL_Complex8>(alpha), B, C)); in add()
168 operation, A, to_mkl_complex<double, MKL_Complex16>(alpha), B, C)); in add()
222 to_mkl_complex<float, MKL_Complex8>(alpha), in mm()
229 to_mkl_complex<float, MKL_Complex8>(beta), in mm()
237 to_mkl_complex<double, MKL_Complex16>(alpha), in mm()
[all …]
|