Searched refs:MatrixR (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_matmul_op.cc | 781 using MatrixR = BasicMatrix<TR>; typedef in tensorflow::SparseMatMul
827 MatrixR* buffer, std::vector<ConstMatrixMapR*>* slices);
835 const DeviceBase::CpuWorkerThreads* thread_pool, MatrixR* buffer);
838 static inline void SliceMatrix(const MatrixR& mat, const int num_rows,
862 using MatrixR = BasicMatrix<TR>; typedef in tensorflow::LibxsmmSparseMatMul
955 using MatrixR = BasicMatrix<TR>; typedef in tensorflow::SparseMatMulOp
1237 const DeviceBase::CpuWorkerThreads* thread_pool, MatrixR* buffer) { in ShuffleMatrix()
1293 const MatrixR& mat, const int num_rows, const int num_slices, in SliceMatrix()
1307 const DeviceBase::CpuWorkerThreads* thread_pool, MatrixR* buffer, in CreateDenseSlices()
1551 MatrixR buffer(buffer_num_rows, N); in Compute()
|