Searched refs:num_preserved_coeffs (Results 1 – 1 of 1) sorted by relevance
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorReductionCuda.h | 114 __global__ void ReductionInitKernel(const CoeffType val, Index num_preserved_coeffs, CoeffType* out… in ReductionInitKernel() argument
117 for (Index i = thread_id; i < num_preserved_coeffs; i += num_threads) { in ReductionInitKernel()
374 …onKernel(Reducer reducer, const Self input, Index num_coeffs_to_reduce, Index num_preserved_coeffs,
387 const Index num_input_blocks = input_col_blocks * num_preserved_coeffs;
394 for (Index i = thread_id; i < num_preserved_coeffs; i += num_threads) {
403 if (row < num_preserved_coeffs) {
446 …alfFloat(Reducer reducer, const Self input, Index num_coeffs_to_reduce, Index num_preserved_coeffs,
458 const Index num_input_blocks = divup<Index>(input_col_blocks * num_preserved_coeffs, 2);
466 for (; i + 1 < num_preserved_coeffs; i += 2*num_threads) {
470 if (i < num_preserved_coeffs) {
[all …]
|