Searched defs:SparseTensorDenseMatMulFunctor (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | sparse_tensor_dense_matmul_op_gpu.cu.cc | 66 struct SparseTensorDenseMatMulFunctor<GPUDevice, T, Tindices, ADJ_A, ADJ_B> { struct 68 Compute(const GPUDevice& d, typename TTypes<T>::Matrix out, in Compute()
|
D | sparse_tensor_dense_matmul_op.h | 30 struct SparseTensorDenseMatMulFunctor { struct
|
D | sparse_tensor_dense_matmul_op.cc | 243 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()
|