Searched refs:num_coeffs_to_preserve (Results 1 - 2 of 2) sorted by relevance
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
H A D | TensorReduction.h | 502 const Index num_coeffs_to_preserve = internal::array_prod(m_dimensions); local 504 data = static_cast<CoeffReturnType*>(m_device.allocate(sizeof(CoeffReturnType) * num_coeffs_to_preserve)); 508 internal::InnerReducer<Self, Op, Device>::run(*this, reducer, m_device, data, num_values_to_reduce, num_coeffs_to_preserve); 525 const Index num_coeffs_to_preserve = internal::array_prod(m_dimensions); local 527 if (num_coeffs_to_preserve < 1024 && num_values_to_reduce > num_coeffs_to_preserve && num_values_to_reduce > 128) { 528 data = static_cast<CoeffReturnType*>(m_device.allocate(sizeof(CoeffReturnType) * num_coeffs_to_preserve)); 536 if (internal::InnerReducer<Self, Op, Device>::run(*this, reducer, m_device, data, num_values_to_reduce, num_coeffs_to_preserve)) { 558 const Index num_coeffs_to_preserve = internal::array_prod(m_dimensions); local 560 if (num_coeffs_to_preserve < 102 [all...] |
H A D | TensorReductionSycl.h | 187 static bool run(const Self& self, Op& reducer, const Eigen::SyclDevice& dev, CoeffReturnType* output, typename Self::Index , typename Self::Index num_coeffs_to_preserve) { argument 194 size_t GRange=num_coeffs_to_preserve; 211 auto output_accessor = dev.template get_sycl_accessor<cl::sycl::access::mode::discard_write>(num_coeffs_to_preserve,cgh, output); 226 if (globalid< static_cast<size_t>(num_coeffs_to_preserve)) {
|
Completed in 368 milliseconds