|
TR-mbed 1.0
|
#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 |
|
inlinestatic |
|
inlinestatic |
|
inlinestatic |
|
static |
|
static |
|
static |
|
static |