Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/sparse/
Dmul_op.cc90 functor::CSRSparseMatrixBatchMulVec<Device, T> csrmul_batch_vec; in Compute()
155 Status CSRSparseMatrixBatchMulVec<GPUDevice, T>::Compute( \
158 extern template struct CSRSparseMatrixBatchMulVec<GPUDevice, T>;
Dkernels.h240 class CSRSparseMatrixBatchMulVec {
242 explicit CSRSparseMatrixBatchMulVec() {} in CSRSparseMatrixBatchMulVec() function
Dkernels_gpu.cu.cc329 CSRSparseMatrixBatchMulVec<GPUDevice, T>::CSRSparseMatrixBatchMulVec() {} \
331 Status CSRSparseMatrixBatchMulVec<GPUDevice, T>::Compute( \