Home
last modified time | relevance | path

Searched refs:costPerCoeff (Results 1 – 25 of 30) sorted by relevance

12

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorEvaluator.h106 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const { in costPerCoeff() function
205 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
269 costPerCoeff(bool vectorized) const {
338 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
340 return m_argImpl.costPerCoeff(vectorized) +
417 costPerCoeff(bool vectorized) const {
419 return m_leftImpl.costPerCoeff(vectorized) +
420 m_rightImpl.costPerCoeff(vectorized) +
516 costPerCoeff(bool vectorized) const {
518 return m_arg1Impl.costPerCoeff(vectorized) +
[all …]
DTensorArgMax.h116 costPerCoeff(bool vectorized) const {
117 return m_impl.costPerCoeff(vectorized) + TensorOpCost(0, 0, 1);
258 costPerCoeff(bool vectorized) const {
261 return m_orig_impl.costPerCoeff(vectorized) +
262 m_impl.costPerCoeff(vectorized) + TensorOpCost(0, 0, compute_cost);
DTensorAssign.h154 costPerCoeff(bool vectorized) const {
158 TensorOpCost left = m_leftImpl.costPerCoeff(vectorized);
159 return m_rightImpl.costPerCoeff(vectorized) +
DTensorConversion.h233 costPerCoeff(bool vectorized) const {
240 return m_impl.costPerCoeff(vectorized) * (SrcCoeffRatio / PacketSize) +
243 return m_impl.costPerCoeff(vectorized) + TensorOpCost(0, 0, cast_cost);
DTensorEvalTo.h155 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
158 return m_impl.costPerCoeff(vectorized) +
DTensorLayoutSwap.h158 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
159 return m_impl.costPerCoeff(vectorized);
DTensorInflation.h203 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
211 return m_impl.costPerCoeff(vectorized) +
DTensorConcatenation.h264 costPerCoeff(bool vectorized) const {
272 m_leftImpl.costPerCoeff(vectorized) +
274 m_rightImpl.costPerCoeff(vectorized) +
DTensorExecutor.h154 device.parallelFor(size, evaluator.costPerCoeff(Vectorizable),
163 size, evaluator.costPerCoeff(Vectorizable), num_threads);
DTensorReverse.h212 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
221 return m_impl.costPerCoeff(vectorized) +
DTensorShuffling.h180 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
184 return m_impl.costPerCoeff(vectorized) +
DTensorMorphing.h145 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
146 return m_impl.costPerCoeff(vectorized);
456 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
457 return m_impl.costPerCoeff(vectorized) + TensorOpCost(0, 0, NumDims);
806 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
807 return m_impl.costPerCoeff(vectorized) + TensorOpCost(0, 0, NumDims);
DTensorPatch.h248 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
252 return m_impl.costPerCoeff(vectorized) +
DTensorCustomOp.h136 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
291 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
DTensorConvolution.h452 costPerCoeff(bool vectorized) const {
462 kernel_size * (m_inputImpl.costPerCoeff(vectorized) +
463 m_kernelImpl.costPerCoeff(vectorized) +
1066 costPerCoeff(bool vectorized) const {
1078 kernel_size * (m_inputImpl.costPerCoeff(vectorized) +
1079 m_kernelImpl.costPerCoeff(vectorized) +
DTensorStriding.h211 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
220 return m_impl.costPerCoeff(vectorized && m_inputStrides[innerDim] == 1) +
DTensorBroadcasting.h352 costPerCoeff(bool vectorized) const {
371 return m_impl.costPerCoeff(vectorized) +
DTensorPadding.h189 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
190 TensorOpCost cost = m_impl.costPerCoeff(vectorized);
DTensorChipping.h245 costPerCoeff(bool vectorized) const {
262 return m_impl.costPerCoeff(vectorized) +
DTensorForcedEval.h147 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
DTensorGenerator.h149 costPerCoeff(bool) const {
DTensorImagePatch.h437 costPerCoeff(bool vectorized) const {
444 return m_impl.costPerCoeff(vectorized) +
DTensorReduction.h259 self.m_impl.costPerCoeff(Vectorizable) +
654 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
660 return m_impl.costPerCoeff(vectorized) * num_values_to_reduce +
DTensorScan.h188 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool) const {
/external/tensorflow/tensorflow/core/kernels/
Dmirror_pad_op.h243 costPerCoeff(bool vectorized) const {
250 return impl_.costPerCoeff(vectorized) +

12