Home
last modified time | relevance | path

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

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorFunctors.h122 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Packet finalizePacket(const Packet& vaccum) const {
123 return vaccum;
126 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE T finalizeBoth(const T saccum, const Packet& vaccum) const {
128 return sum_op(saccum, predux(vaccum));
172 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Packet finalizePacket(const Packet& vaccum) const {
173 return pdiv(vaccum, pset1<Packet>(packetCount_));
176 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE T finalizeBoth(const T saccum, const Packet& vaccum) const {
178 …return sum_op(saccum, predux(vaccum)) / (scalarCount_ + packetCount_ * unpacket_traits<Packet>::si…
244 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Packet finalizePacket(const Packet& vaccum) const {
245 return vaccum;
[all …]
/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op.cc98 auto vaccum = Eigen::internal::pset1<Packet>(static_cast<T>(0)); in Run() local
111 vaccum = in Run()
112 Eigen::internal::pmadd<Packet>(filter_block, data_block, vaccum); in Run()
115 Eigen::internal::pstoreu<T>(output + base_output_index + i, vaccum); in Run()
119 auto vaccum = Eigen::internal::pset1<Packet>(static_cast<T>(0)); in Run() local
127 vaccum = in Run()
128 Eigen::internal::pmadd<Packet>(filter_block, data_block, vaccum); in Run()
132 Eigen::internal::pstoreu<T>(out_buf, vaccum); in Run()
Ddepthwise_conv_grad_op.cc304 auto vaccum = Eigen::internal::pset1<Packet>(static_cast<T>(0)); in ComputeBackpropInput() local
313 vaccum = Eigen::internal::pmadd<Packet>(filter_block, data_block, vaccum); in ComputeBackpropInput()
317 Eigen::internal::pstoreu<T>(output + base_output_index + i, vaccum); in ComputeBackpropInput()
320 Eigen::internal::pstoreu<T>(out_buffer + i, vaccum); in ComputeBackpropInput()
325 auto vaccum = Eigen::internal::pset1<Packet>(static_cast<T>(0)); in ComputeBackpropInput() local
331 vaccum = Eigen::internal::pmadd<Packet>(filter_block, data_block, vaccum); in ComputeBackpropInput()
335 Eigen::internal::pstoreu<T>(out_buf, vaccum); in ComputeBackpropInput()
Deigen_pooling.h353 finalizePacket(const Packet& vaccum) const { in finalizePacket()
354 return pdiv(vaccum, packetCount_); in finalizePacket()
358 finalizeBoth(const T saccum, const Packet& vaccum) const { in finalizeBoth()
359 return (saccum + predux(vaccum)) / (scalarCount_ + predux(packetCount_)); in finalizeBoth()