blob: 674123a2058331d31d56b201a36bc2d8f33f034e (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
--- a/tensorflow/core/kernels/sparse/transpose_op.cc 2022-10-24 07:47:20.696262052 +0200
+++ b/tensorflow/core/kernels/sparse/transpose_op.cc 2022-10-23 20:50:10.801620356 +0200
@@ -208,6 +208,13 @@
return OkStatus();
}
+#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
+template struct CSRSparseMatrixTranspose<GPUDevice, float>;
+template struct CSRSparseMatrixTranspose<GPUDevice, double>;
+template struct CSRSparseMatrixTranspose<GPUDevice, std::complex<float>>;
+template struct CSRSparseMatrixTranspose<GPUDevice, std::complex<double>>;
+#endif
+
// CPU kernel for transposing a single component of a CSR SparseMatrix.
template <typename T>
struct CSRSparseMatrixTransposeComponent<CPUDevice, T> {
|