#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 Public Attributes | |
static const int | kDeviceCyclesPerComputeCycle = 1 |
static const int | kPerThreadCycles = 100000 |
static const int | kStartupCycles = 100000 |
static const int | kTaskSize = 40000 |
Static Private Member Functions | |
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double | totalCost (double output_size, const TensorOpCost &cost_per_coeff) |
Definition at line 161 of file TensorCostModel.h.
|
inlinestatic |
Definition at line 174 of file TensorCostModel.h.
|
inlinestatic |
Definition at line 184 of file TensorCostModel.h.
|
inlinestaticprivate |
Definition at line 190 of file TensorCostModel.h.
|
static |
Definition at line 164 of file TensorCostModel.h.
|
static |
Definition at line 168 of file TensorCostModel.h.
|
static |
Definition at line 167 of file TensorCostModel.h.
|
static |
Definition at line 169 of file TensorCostModel.h.