Home
last modified time | relevance | path

Searched refs:reducePacket (Results 1 – 4 of 4) sorted by relevance

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorReductionCuda.h87 reducer.reducePacket(accum, reinterpret_cast<half2*>(&newval)); in atomicReduce()
95 reducer.reducePacket(accum, reinterpret_cast<half2*>(&newval)); in atomicReduce()
242 reducer.reducePacket(val, &accum); in FullReductionKernelHalfFloat()
247 reducer.reducePacket(__shfl_down(accum, offset, warpSize), &accum); in FullReductionKernelHalfFloat()
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), &r…
512 …reducer.reducePacket(input.m_impl.template packet<Unaligned>((row + 1)* num_coeffs_to_reduce + col…
[all …]
DTensorFunctors.h106 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 {
DTensorReduction.h173 reducer.reducePacket(self.m_impl.template packet<Unaligned>(firstIndex + j), &p);
206 reducer.reducePacket(self.m_impl.template packet<Unaligned>(input), accum);
/external/tensorflow/tensorflow/core/kernels/
Deigen_pooling.h331 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reducePacket(const Packet& p, in reducePacket() function