HomeSort by relevance Sort by last modified time
    Searched refs:num_coeffs_to_preserve (Results 1 - 2 of 2) sorted by null

  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
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) {
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)) {
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 < 1024 && num_values_to_reduce > num_coeffs_to_preserve && num_values_to_reduce > 32)
    [all...]

Completed in 85 milliseconds