Home
last modified time | relevance | path

Searched refs:vaccum (Results 1 – 5 of 5) 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/
Dscan_ops.h107 finalizePacket(const Packet& vaccum) const { in finalizePacket()
108 return vaccum; in finalizePacket()
113 finalizeBoth(const T saccum, const Packet& vaccum) const { in finalizeBoth()
131 auto ma = max_reducer.finalizeBoth(saccum, vaccum); in finalizeBoth()
133 exp(saccum - ma), pexp(psub(vaccum, pset1(ma))))), in finalizeBoth()
Ddepthwise_conv_op.cc103 auto vaccum = Eigen::internal::pset1<Packet>(static_cast<T>(0)); in Run() local
116 vaccum = in Run()
117 Eigen::internal::pmadd<Packet>(filter_block, data_block, vaccum); in Run()
120 Eigen::internal::pstoreu<T>(output + base_output_index + i, vaccum); in Run()
124 auto vaccum = Eigen::internal::pset1<Packet>(static_cast<T>(0)); in Run() local
132 vaccum = in Run()
133 Eigen::internal::pmadd<Packet>(filter_block, data_block, vaccum); in Run()
137 Eigen::internal::pstoreu<T>(out_buf, vaccum); in Run()
Ddepthwise_conv_grad_op.cc316 auto vaccum = Eigen::internal::pset1<Packet>(static_cast<T>(0)); in ComputeBackpropInput() local
325 vaccum = Eigen::internal::pmadd<Packet>(filter_block, data_block, vaccum); in ComputeBackpropInput()
329 Eigen::internal::pstoreu<T>(output + base_output_index + i, vaccum); in ComputeBackpropInput()
332 Eigen::internal::pstoreu<T>(out_buffer + i, vaccum); in ComputeBackpropInput()
337 auto vaccum = Eigen::internal::pset1<Packet>(static_cast<T>(0)); in ComputeBackpropInput() local
343 vaccum = Eigen::internal::pmadd<Packet>(filter_block, data_block, vaccum); in ComputeBackpropInput()
347 Eigen::internal::pstoreu<T>(out_buf, vaccum); in ComputeBackpropInput()
Deigen_pooling.h355 finalizePacket(const Packet& vaccum) const { in finalizePacket()
356 return pdiv(vaccum, packetCount_); in finalizePacket()
360 finalizeBoth(const T saccum, const Packet& vaccum) const { in finalizeBoth()
361 return (saccum + predux(vaccum)) / (scalarCount_ + predux(packetCount_)); in finalizeBoth()