OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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