Searched refs:num_out_rows (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reduction_gpu_kernels.cu.h | 461 const int num_out_rows, int num_rows,
472 in_[plane * elems_per_plane + i * (num_out_rows * num_cols) + col];
478 offset = 2 * kUnroll * (num_out_rows * num_cols);
508 const int num_out_rows = max(1, num_rows / (2 * kUnroll));
509 const int plane = gid / (num_out_rows * num_cols);
510 const int col = gid % (num_out_rows * num_cols);
517 sum = ComputeSum<value_type, IN_T, Op>(in_, plane, num_out_rows, num_rows,
521 sum = ComputeSum<value_type, value_type*, Op>(in_, plane, num_out_rows,
524 out_[plane * num_out_rows * num_cols + col] = sum;
574 int num_out_rows;
[all …]
|
| D | sparse_matmul_op.cc | 1280 int num_out_rows = (slice_num_cols + N - 1) / N * slice_num_rows; in ShuffleMatrix() local
1281 DCHECK_LE(num_out_rows, buffer->dimension(0)); in ShuffleMatrix()
1283 end = start + num_out_rows / i; in ShuffleMatrix()
1285 num_out_rows -= (end - start); in ShuffleMatrix()
|