Fix tensor documentation.

This commit is contained in:
Antonio Sánchez
2025-02-05 17:36:00 +00:00
parent 74264c391a
commit c079ee5e44
41 changed files with 272 additions and 329 deletions

View File

@@ -15,14 +15,6 @@
namespace Eigen {
/** \class TensorEvaluator
* \ingroup CXX11_Tensor_Module
*
* \brief A cost model used to limit the number of threads used for evaluating
* tensor expression.
*
*/
// Class storing the cost of evaluating a tensor expression in terms of the
// estimated number of operand bytes loads, bytes stored, and compute cycles.
class TensorOpCost {
@@ -137,6 +129,13 @@ class TensorOpCost {
// TODO(rmlarsen): Implement a policy that chooses an "optimal" number of theads
// in [1:max_threads] instead of just switching multi-threading off for small
// work units.
/**
* \ingroup CXX11_Tensor_Module
*
* \brief A cost model used to limit the number of threads used for evaluating
* tensor expression.
*
*/
template <typename Device>
class TensorCostModel {
public: