Home
last modified time | relevance | path

Searched refs:packetCount_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_pooling.h288 packetCount_ = pset1<Packet>(T(0.0)); in AvgPoolMeanReducer()
344 packetCount_ = padd<Packet>( in reducePacketWithType()
345 packetCount_, psel(pset1<Packet>(1), pset1<Packet>(0), skip_mask)); in reducePacketWithType()
356 return pdiv(vaccum, packetCount_); in finalizePacket()
361 return (saccum + predux(vaccum)) / (scalarCount_ + predux(packetCount_)); in finalizeBoth()
368 Packet packetCount_; member
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorFunctors.h147 MeanReducer() : scalarCount_(0), packetCount_(0) { }
157 packetCount_++;
173 return pdiv(vaccum, pset1<Packet>(packetCount_));
178 …return sum_op(saccum, predux(vaccum)) / (scalarCount_ + packetCount_ * unpacket_traits<Packet>::si…
183 DenseIndex packetCount_;