Searched refs:num_output_rows (Results 1 – 2 of 2) sorted by relevance
174 const int64 num_output_rows = output_shape_vec(row_dim); in Compute() local182 num_output_rows * (avg_nnz_per_row_a * avg_nnz_per_row_b); in Compute()216 TensorShape({(num_output_rows + 1) * batch_size})); in Compute()234 output_matrix.outerIndexPtr() + num_output_rows + 1, in Compute()235 output_row_ptr_ptr + batch_idx * (num_output_rows + 1)); in Compute()
237 int num_output_rows = output_matrix_shapes[i].dims() >= 1 in ComputeTensorSlice() local246 num_output_rows, num_output_cols); in ComputeTensorSlice()