Searched refs:scalar_mod_op (Results 1 – 4 of 4) sorted by relevance
21 struct scalar_mod_op { struct22 EIGEN_DEVICE_FUNC scalar_mod_op(const Scalar& divisor) : m_divisor(divisor) {} in scalar_mod_op() argument27 struct functor_traits<scalar_mod_op<Scalar> > argument
46 return internal::functor_traits<internal::scalar_mod_op<ArgType> >::Cost; in ModCost()
284 EIGEN_STRONG_INLINE const TensorCwiseUnaryOp<internal::scalar_mod_op<Scalar>, const Derived>287 return unaryExpr(internal::scalar_mod_op<Scalar>(rhs));
613 elementwise_ops_.emplace("FloorMod", EIGEN_COST(scalar_mod_op<float>)); in OpLevelCostEstimator()623 elementwise_ops_.emplace("Mod", EIGEN_COST(scalar_mod_op<float>)); in OpLevelCostEstimator()639 elementwise_ops_.emplace("TruncateMod", EIGEN_COST(scalar_mod_op<float>)); in OpLevelCostEstimator()