diff options
Diffstat (limited to 'tensorflow/core/kernels/sparse_tensor_dense_matmul_op.h')
| -rw-r--r-- | tensorflow/core/kernels/sparse_tensor_dense_matmul_op.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/sparse_tensor_dense_matmul_op.h b/tensorflow/core/kernels/sparse_tensor_dense_matmul_op.h index bcb836367b..e707743f78 100644 --- a/tensorflow/core/kernels/sparse_tensor_dense_matmul_op.h +++ b/tensorflow/core/kernels/sparse_tensor_dense_matmul_op.h @@ -25,11 +25,12 @@ namespace tensorflow { namespace functor { -template <typename Device, typename T, bool ADJ_A, bool ADJ_B> +template <typename Device, typename T, typename Tindices, bool ADJ_A, + bool ADJ_B> struct SparseTensorDenseMatMulFunctor { static EIGEN_ALWAYS_INLINE Status Compute(const Device& d, typename TTypes<T>::Matrix out, - TTypes<int64>::ConstMatrix a_indices, + typename TTypes<Tindices>::ConstMatrix a_indices, typename TTypes<T>::ConstVec a_values, typename TTypes<T>::ConstMatrix b, typename TTypes<T>::Vec scratch); }; |