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

  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorReductionCuda.h 87 reducer.reducePacket(accum, reinterpret_cast<half2*>(&newval));
95 reducer.reducePacket(accum, reinterpret_cast<half2*>(&newval));
242 reducer.reducePacket(val, &accum);
247 reducer.reducePacket(__shfl_down(accum, offset, warpSize), &accum);
492 reducer.reducePacket(val1, &reduced_val1);
494 reducer.reducePacket(val2, &reduced_val2);
500 reducer.reducePacket(val1, &reduced_val1);
503 reducer.reducePacket(val2, &reduced_val2);
511 reducer.reducePacket(input.m_impl.template packet<Unaligned>(row * num_coeffs_to_reduce + col), &reduced_val1);
512 reducer.reducePacket(input.m_impl.template packet<Unaligned>((row + 1)* num_coeffs_to_reduce + col), &reduced_val2)
    [all...]
TensorFunctors.h 106 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reducePacket(const Packet& p, Packet* accum) const {
155 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reducePacket(const Packet& p, Packet* accum) {
230 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reducePacket(const Packet& p, Packet* accum) const {
271 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reducePacket(const Packet& p, Packet* accum) const {
313 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reducePacket(const Packet& p, Packet* accum) const {
TensorReduction.h 173 reducer.reducePacket(self.m_impl.template packet<Unaligned>(firstIndex + j), &p);
206 reducer.reducePacket(self.m_impl.template packet<Unaligned>(input), accum);
    [all...]

Completed in 53 milliseconds