Home
last modified time | relevance | path

Searched defs:SparseTensorDenseMatMulFunctor (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_tensor_dense_matmul_op_gpu.cu.cc66 struct SparseTensorDenseMatMulFunctor<GPUDevice, T, Tindices, ADJ_A, ADJ_B> { struct
68 Compute(const GPUDevice& d, typename TTypes<T>::Matrix out, in Compute()
Dsparse_tensor_dense_matmul_op.h30 struct SparseTensorDenseMatMulFunctor { struct
Dsparse_tensor_dense_matmul_op.cc243 struct SparseTensorDenseMatMulFunctor<CPUDevice, T, Tindices, ADJ_A, ADJ_B> { struct
245 static const std::size_t kNumVectorize = 32;
247 static Status Compute(const CPUDevice& d, typename TTypes<T>::Matrix out, in Compute()