Home | History | Annotate | Download | only in Tensor

Lines Matching refs:num_coeffs

223     const typename Self::Index num_coeffs = array_prod(self.m_impl.dimensions());
224 *output = InnerMostDimReducer<Self, Op, Vectorizable>::reduce(self, 0, num_coeffs, reducer);
253 const Index num_coeffs = array_prod(self.m_impl.dimensions());
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) {
284 self, numblocks * blocksize, num_coeffs - numblocks * blocksize,