Home
last modified time | relevance | path

Searched refs:mat_rows (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_matvec.h46 int64 mat_rows = rows; in MatVec() local
50 std::swap(mat_rows, mat_cols); in MatVec()
53 auto mat = MatrixMap(mat_buf, mat_rows, mat_cols); in MatVec()
/external/tensorflow/tensorflow/core/kernels/
Dsparse_matmul_op.cc181 const int mat_rows = Transpose ? mat.dimension(1) : mat.dimension(0); in Initialize() local
183 DCHECK_LE(num_rows, mat_rows); in Initialize()