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

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
runtime_matvec.h 47 int64 mat_cols = cols; local
50 std::swap(mat_rows, mat_cols);
53 auto mat = MatrixMap(mat_buf, mat_rows, mat_cols);
  /external/mesa3d/src/compiler/nir/
nir_gather_info.c 172 unsigned mat_cols = 1; local
176 mat_cols = glsl_get_matrix_columns(glsl_without_array(type));
186 if (offset >= num_elems * elem_width * mat_cols) {
  /external/tensorflow/tensorflow/core/kernels/
sparse_matmul_op.cc 182 const int mat_cols = Transpose ? mat.dimension(0) : mat.dimension(1); local
184 DCHECK_LE(num_cols + col_offset, mat_cols);
    [all...]

Completed in 307 milliseconds