HomeSort by relevance Sort by last modified time
    Searched refs:mat_rows (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
runtime_matvec.h 46 int64 mat_rows = rows; local
50 std::swap(mat_rows, mat_cols);
53 auto mat = MatrixMap(mat_buf, mat_rows, mat_cols);
  /external/tensorflow/tensorflow/core/kernels/
sparse_matmul_op.cc 181 const int mat_rows = Transpose ? mat.dimension(1) : mat.dimension(0); local
183 DCHECK_LE(num_rows, mat_rows);
    [all...]

Completed in 3856 milliseconds