Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/sparse/
Dmat_mul_op.cc729 struct GPUDataType;
736 struct GPUDataType<Eigen::half> { struct
743 struct GPUDataType<float> { struct
750 struct GPUDataType<std::complex<float>> { struct
757 struct GPUDataType<double> { struct
764 struct GPUDataType<std::complex<double>> { struct
843 GPUDataType<T>::type)); in Compute()
847 GPUDataType<T>::type, CUSPARSE_ORDER_COL)); in Compute()
850 cusparseCreateDnMat(&matC, m, n, ldc, c.data(), GPUDataType<T>::type, in Compute()