Searched refs:MatrixR (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_matmul_op.cc | 782 using MatrixR = BasicMatrix<TR>; typedef in tensorflow::SparseMatMul
828 MatrixR* buffer, std::vector<ConstMatrixMapR*>* slices);
836 const DeviceBase::CpuWorkerThreads* thread_pool, MatrixR* buffer);
839 static inline void SliceMatrix(const MatrixR& mat, const int num_rows,
863 using MatrixR = BasicMatrix<TR>; typedef in tensorflow::LibxsmmSparseMatMul
956 using MatrixR = BasicMatrix<TR>; typedef in tensorflow::SparseMatMulOp
1238 const DeviceBase::CpuWorkerThreads* thread_pool, MatrixR* buffer) { in ShuffleMatrix()
1292 const MatrixR& mat, const int num_rows, const int num_slices, in SliceMatrix()
1306 const DeviceBase::CpuWorkerThreads* thread_pool, MatrixR* buffer, in CreateDenseSlices()
1554 MatrixR buffer(buffer_num_rows, N); in Compute()
|