HomeSort by relevance Sort by last modified time
    Searched refs:num_coeffs (Results 1 - 5 of 5) 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...]
  /external/mesa3d/src/gallium/auxiliary/gallivm/
lp_bld_arit.h 311 unsigned num_coeffs);
lp_bld_arit.c     [all...]
  /prebuilts/gcc/linux-x86/host/x86_64-w64-mingw32-4.8/x86_64-w64-mingw32/include/
d3dx9mesh.h 621 STDMETHOD(ApplyGuttersFloat)(THIS_ FLOAT *data_in, UINT num_coeffs, UINT width, UINT height) PURE;
    [all...]

Completed in 129 milliseconds