OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:num_coeffs
(Results
1 - 2
of
2
) sorted by null
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorReductionCuda.h
125
__global__ void FullReductionKernel(Reducer reducer, const Self input, Index
num_coeffs
,
161
Index max_iter = numext::mini<Index>(
num_coeffs
- first_index, NumPerThread*BlockSize);
164
eigen_assert(index <
num_coeffs
);
191
__global__ void ReductionInitFullReduxKernelHalfFloat(Reducer reducer, const Self input, Index
num_coeffs
, half2* scratch) {
194
if (
num_coeffs
% 2 != 0) {
195
half last = input.m_impl.coeff(
num_coeffs
-1);
204
__global__ void ReductionInitKernelHalfFloat(Reducer reducer, const Self input, Index
num_coeffs
, half* output) {
207
const Index num_packets =
num_coeffs
/ 2;
212
if (thread_id == 0 &&
num_coeffs
% 2 != 0) {
213
output[
num_coeffs
-1] = reducer.initialize()
361
const Index
num_coeffs
= array_prod(self.m_impl.dimensions());
local
558
const Index
num_coeffs
=
num_coeffs
_to_reduce * num_preserved_vals;
local
604
const Index
num_coeffs
=
num_coeffs
_to_reduce * num_preserved_vals;
local
651
const Index
num_coeffs
= array_prod(self.m_impl.dimensions());
local
717
const Index
num_coeffs
=
num_coeffs
_to_reduce * num_preserved_vals;
local
[
all
...]
TensorReduction.h
223
const typename Self::Index
num_coeffs
= array_prod(self.m_impl.dimensions());
local
224
*output = InnerMostDimReducer<Self, Op, Vectorizable>::reduce(self, 0,
num_coeffs
, reducer);
253
const Index
num_coeffs
= array_prod(self.m_impl.dimensions());
local
254
if (
num_coeffs
== 0) {
263
num_coeffs
, cost, device.numThreads());
266
InnerMostDimReducer<Self, Op, Vectorizable>::reduce(self, 0,
num_coeffs
, reducer);
270
std::floor<Index>(static_cast<float>(
num_coeffs
) / num_threads);
271
const Index numblocks = blocksize > 0 ?
num_coeffs
/ blocksize : 0;
272
eigen_assert(
num_coeffs
>= numblocks * blocksize);
282
if (numblocks * blocksize <
num_coeffs
) {
[
all
...]
Completed in 88 milliseconds