TR-mbed 1.0
Loading...
Searching...
No Matches
Static Public Member Functions | Static Public Attributes | List of all members
Eigen::TensorCostModel< Device > Class Template Reference

#include <TensorCostModel.h>

Static Public Member Functions

static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE int numThreads (double output_size, const TensorOpCost &cost_per_coeff, int max_threads)
 
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double taskSize (double output_size, const TensorOpCost &cost_per_coeff)
 
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double totalCost (double output_size, const TensorOpCost &cost_per_coeff)
 

Static Public Attributes

static const int kDeviceCyclesPerComputeCycle = 1
 
static const int kStartupCycles = 100000
 
static const int kPerThreadCycles = 100000
 
static const int kTaskSize = 40000
 

Member Function Documentation

◆ numThreads()

template<typename Device >
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE int Eigen::TensorCostModel< Device >::numThreads ( double  output_size,
const TensorOpCost cost_per_coeff,
int  max_threads 
)
inlinestatic

◆ taskSize()

template<typename Device >
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double Eigen::TensorCostModel< Device >::taskSize ( double  output_size,
const TensorOpCost cost_per_coeff 
)
inlinestatic

◆ totalCost()

template<typename Device >
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double Eigen::TensorCostModel< Device >::totalCost ( double  output_size,
const TensorOpCost cost_per_coeff 
)
inlinestatic

Member Data Documentation

◆ kDeviceCyclesPerComputeCycle

template<typename Device >
const int Eigen::TensorCostModel< Device >::kDeviceCyclesPerComputeCycle = 1
static

◆ kPerThreadCycles

template<typename Device >
const int Eigen::TensorCostModel< Device >::kPerThreadCycles = 100000
static

◆ kStartupCycles

template<typename Device >
const int Eigen::TensorCostModel< Device >::kStartupCycles = 100000
static

◆ kTaskSize

template<typename Device >
const int Eigen::TensorCostModel< Device >::kTaskSize = 40000
static

The documentation for this class was generated from the following file: