Home
last modified time | relevance | path

Searched refs:TensorOpCost (Results 1 – 25 of 46) sorted by relevance

12

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorCostModel.h25 class TensorOpCost {
55 TensorOpCost() : bytes_loaded_(0), bytes_stored_(0), compute_cycles_(0) {} in TensorOpCost() function
57 TensorOpCost(double bytes_loaded, double bytes_stored, double compute_cycles) in TensorOpCost() function
63 TensorOpCost(double bytes_loaded, double bytes_stored, double compute_cycles, in TensorOpCost() function
97 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost cwiseMin( in cwiseMin()
98 const TensorOpCost& rhs) const { in cwiseMin()
102 return TensorOpCost(bytes_loaded, bytes_stored, compute_cycles); in cwiseMin()
106 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost cwiseMax( in cwiseMax()
107 const TensorOpCost& rhs) const { in cwiseMax()
111 return TensorOpCost(bytes_loaded, bytes_stored, compute_cycles); in cwiseMax()
[all …]
DTensorBroadcasting.h351 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost
353 double compute_cost = TensorOpCost::AddCost<Index>();
356 compute_cost += TensorOpCost::DivCost<Index>();
359 TensorOpCost::MulCost<Index>() + TensorOpCost::AddCost<Index>();
362 compute_cost += TensorOpCost::MulCost<Index>() +
363 TensorOpCost::ModCost<Index>() +
364 TensorOpCost::AddCost<Index>();
368 TensorOpCost::MulCost<Index>() + TensorOpCost::AddCost<Index>();
372 TensorOpCost(0, 0, compute_cost, vectorized, PacketSize);
DTensorPadding.h189 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
190 TensorOpCost cost = m_impl.costPerCoeff(vectorized);
238 void updateCostPerDimension(TensorOpCost& cost, int i, bool first) const {
246 cost += TensorOpCost(0, 0, 2 * TensorOpCost::AddCost<Index>() +
247 reduction * (1 * TensorOpCost::AddCost<Index>()));
249 cost += TensorOpCost(0, 0, 2 * TensorOpCost::AddCost<Index>() +
250 2 * TensorOpCost::MulCost<Index>() +
251 reduction * (2 * TensorOpCost::MulCost<Index>() +
252 1 * TensorOpCost::DivCost<Index>()));
DTensorInflation.h203 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
204 const double compute_cost = NumDims * (3 * TensorOpCost::DivCost<Index>() +
205 3 * TensorOpCost::MulCost<Index>() +
206 2 * TensorOpCost::AddCost<Index>());
210 return TensorOpCost();
212 TensorOpCost(sizeof(CoeffReturnType) * input_size / output_size, 0,
DTensorEvaluator.h106 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const { in costPerCoeff()
107 return TensorOpCost(sizeof(CoeffReturnType), 0, 0, vectorized, in costPerCoeff()
205 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
206 return TensorOpCost(sizeof(CoeffReturnType), 0, 0, vectorized,
268 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost
270 return TensorOpCost(sizeof(CoeffReturnType), 0, 0, vectorized,
338 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
341 TensorOpCost(0, 0, functor_cost, vectorized, PacketSize);
416 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost
421 TensorOpCost(0, 0, functor_cost, vectorized, PacketSize);
[all …]
DTensorReverse.h212 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
213 double compute_cost = NumDims * (2 * TensorOpCost::AddCost<Index>() +
214 2 * TensorOpCost::MulCost<Index>() +
215 TensorOpCost::DivCost<Index>());
218 compute_cost += 2 * TensorOpCost::AddCost<Index>();
222 TensorOpCost(0, 0, compute_cost, false /* vectorized */, PacketSize);
DTensorArgMax.h115 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost
117 return m_impl.costPerCoeff(vectorized) + TensorOpCost(0, 0, 1);
257 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost
260 … (m_return_dim < 0 ? 0.0 : (TensorOpCost::ModCost<Index>() + TensorOpCost::DivCost<Index>()));
262 m_impl.costPerCoeff(vectorized) + TensorOpCost(0, 0, compute_cost);
DTensorChipping.h244 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost
251 cost += TensorOpCost::MulCost<Index>() + TensorOpCost::AddCost<Index>();
256 cost += TensorOpCost::AddCost<Index>();
258 cost += 3 * TensorOpCost::MulCost<Index>() + TensorOpCost::DivCost<Index>() +
259 3 * TensorOpCost::AddCost<Index>();
263 TensorOpCost(0, 0, cost, vectorized, PacketSize);
DTensorGenerator.h148 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost
152 return TensorOpCost(0, 0, TensorOpCost::AddCost<Scalar>() +
153 TensorOpCost::MulCost<Scalar>());
DTensorShuffling.h180 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
181 const double compute_cost = NumDims * (2 * TensorOpCost::AddCost<Index>() +
182 2 * TensorOpCost::MulCost<Index>() +
183 TensorOpCost::DivCost<Index>());
185 TensorOpCost(0, 0, compute_cost, false /* vectorized */, PacketSize);
DTensorAssign.h153 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost
158 TensorOpCost left = m_leftImpl.costPerCoeff(vectorized);
160 TensorOpCost(
163 TensorOpCost(0, sizeof(CoeffReturnType), 0, vectorized, PacketSize);
DTensorPatch.h248 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
249 const double compute_cost = NumDims * (TensorOpCost::DivCost<Index>() +
250 TensorOpCost::MulCost<Index>() +
251 2 * TensorOpCost::AddCost<Index>());
253 TensorOpCost(0, 0, compute_cost, vectorized, PacketSize);
DTensorStriding.h211 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
212 double compute_cost = (NumDims - 1) * (TensorOpCost::AddCost<Index>() +
213 TensorOpCost::MulCost<Index>() +
214 TensorOpCost::DivCost<Index>()) +
215 TensorOpCost::MulCost<Index>();
222 TensorOpCost(0, 0, compute_cost, vectorized, PacketSize);
DTensorConvolution.h451 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost
456 TensorOpCost::AddCost<Scalar>() + TensorOpCost::MulCost<Scalar>();
459 (2 * TensorOpCost::AddCost<Index>() + 2 * TensorOpCost::MulCost<Index>() +
460 TensorOpCost::DivCost<Index>());
461 return TensorOpCost(0, 0, firstIndex_compute_cost, vectorized, PacketSize) +
464 TensorOpCost(0, 0, convolve_compute_cost, vectorized,
1065 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost
1072 TensorOpCost::AddCost<Scalar>() + TensorOpCost::MulCost<Scalar>();
1075 (2 * TensorOpCost::AddCost<Index>() + 2 * TensorOpCost::MulCost<Index>() +
1076 TensorOpCost::DivCost<Index>());
[all …]
DTensorConcatenation.h263 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost
265 const double compute_cost = NumDims * (2 * TensorOpCost::AddCost<Index>() +
266 2 * TensorOpCost::MulCost<Index>() +
267 TensorOpCost::DivCost<Index>() +
268 TensorOpCost::ModCost<Index>());
275 TensorOpCost(0, 0, compute_cost);
DTensorConversion.h232 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost
234 const double cast_cost = TensorOpCost::CastCost<SrcType, TargetType>();
241 TensorOpCost(0, 0, TgtCoeffRatio * (cast_cost / PacketSize));
243 return m_impl.costPerCoeff(vectorized) + TensorOpCost(0, 0, cast_cost);
DTensorCustomOp.h136 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
138 return TensorOpCost(sizeof(CoeffReturnType), 0, 0, vectorized, PacketSize);
291 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(bool vectorized) const {
293 return TensorOpCost(sizeof(CoeffReturnType), 0, 0, vectorized, PacketSize);
/external/tensorflow/tensorflow/core/kernels/
Dparameterized_truncated_normal_op.cc278 (Eigen::TensorOpCost::AddCost<T>() + in operator ()()
279 Eigen::TensorOpCost::MulCost<T>()) * in operator ()()
282 + Eigen::TensorOpCost::AddCost<T>() + in operator ()()
283 Eigen::TensorOpCost::MulCost<T>() + in operator ()()
287 + Eigen::TensorOpCost::MulCost<T>() * 4 + in operator ()()
290 + Eigen::TensorOpCost::AddCost<T>(); in operator ()()
296 uniformSampleCost + Eigen::TensorOpCost::MulCost<T>() + in operator ()()
297 Eigen::TensorOpCost::AddCost<T>() + in operator ()()
298 Eigen::TensorOpCost::MulCost<T>() * 2 + in operator ()()
299 Eigen::TensorOpCost::AddCost<T>() + uniformSampleCost + in operator ()()
[all …]
Dtranspose_functor_cpu.cc60 (conjugate ? 1 : 0) + ndims * (Eigen::TensorOpCost::DivCost<int64>() + in TransposeSimple()
61 2 * Eigen::TensorOpCost::MulCost<int64>() + in TransposeSimple()
62 2 * Eigen::TensorOpCost::AddCost<int64>()); in TransposeSimple()
63 Eigen::TensorOpCost cost(/*bytes_loaded=*/sizeof(T), in TransposeSimple()
Dmatrix_triangular_solve_op.cc72 (Eigen::TensorOpCost::AddCost<Scalar>() + in GetCostPerUnit()
73 Eigen::TensorOpCost::MulCost<Scalar>()); in GetCostPerUnit()
164 (Eigen::TensorOpCost::AddCost<Scalar>() + in GetCostPerUnit()
165 Eigen::TensorOpCost::MulCost<Scalar>()); in GetCostPerUnit()
Dtridiagonal_solve_op.cc70 const double add_cost = Eigen::TensorOpCost::AddCost<Scalar>(); in GetCostPerUnit()
71 const double mult_cost = Eigen::TensorOpCost::MulCost<Scalar>(); in GetCostPerUnit()
72 const double div_cost = Eigen::TensorOpCost::DivCost<Scalar>(); in GetCostPerUnit()
Dcrop_and_resize_op.cc310 depth * (Eigen::TensorOpCost::AddCost<float>() * 6 + in operator ()()
311 Eigen::TensorOpCost::MulCost<float>() * 3 + in operator ()()
312 Eigen::TensorOpCost::CastCost<T, float>() * 4) + in operator ()()
313 (Eigen::TensorOpCost::AddCost<float>() * 2 + in operator ()()
314 Eigen::TensorOpCost::AddCost<float>() * 3); in operator ()()
316 cost_per_pixel = depth * Eigen::TensorOpCost::CastCost<T, float>() + in operator ()()
317 Eigen::TensorOpCost::AddCost<float>() * 4 + in operator ()()
318 Eigen::TensorOpCost::MulCost<float>() * 4; in operator ()()
Ddense_update_functor.cc44 Eigen::TensorOpCost(.1, // chosen to force large chunks in operator ()()
66 Eigen::TensorOpCost(estimated_string_size, estimated_string_size, 0), in operator ()()
Dmirror_pad_op.h236 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost
241 const double compute_cost = Dims * (7 * TensorOpCost::AddCost<Index>() +
242 2 * TensorOpCost::MulCost<Index>() +
243 TensorOpCost::DivCost<Index>());
245 TensorOpCost(1, 0, compute_cost, vectorized, kPacketSize);
/external/tensorflow/tensorflow/contrib/seq2seq/kernels/
Dbeam_search_ops.cc157 Eigen::TensorOpCost::DivCost<int32>() + in operator ()()
158 6 * Eigen::TensorOpCost::AddCost<int32>() + in operator ()()
159 2 * max_time * (5 * Eigen::TensorOpCost::AddCost<int32>()); in operator ()()

12