#include <TensorCostModel.h>
Public Member Functions | |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double | bytes_loaded () const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double | bytes_stored () const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double | compute_cycles () const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost | cwiseMax (const TensorOpCost &rhs) const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost | cwiseMin (const TensorOpCost &rhs) const |
EIGEN_DEVICE_FUNC void | dropMemoryCost () |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost & | operator*= (double rhs) |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost & | operator+= (const TensorOpCost &rhs) |
EIGEN_DEVICE_FUNC | TensorOpCost () |
EIGEN_DEVICE_FUNC | TensorOpCost (double bytes_loaded, double bytes_stored, double compute_cycles) |
EIGEN_DEVICE_FUNC | TensorOpCost (double bytes_loaded, double bytes_stored, double compute_cycles, bool vectorized, double packet_size) |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE double | total_cost (double load_cost, double store_cost, double compute_cost) const |
Static Public Member Functions | |
template<typename ArgType > | |
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE int | AddCost () |
template<typename SrcType , typename TargetType > | |
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE int | CastCost () |
template<typename ArgType > | |
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE int | DivCost () |
template<typename ArgType > | |
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE int | ModCost () |
template<typename ArgType > | |
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE int | MulCost () |
Private Attributes | |
double | bytes_loaded_ |
double | bytes_stored_ |
double | compute_cycles_ |
Friends | |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE friend TensorOpCost | operator* (TensorOpCost lhs, double rhs) |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE friend TensorOpCost | operator* (double lhs, TensorOpCost rhs) |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE friend TensorOpCost | operator+ (TensorOpCost lhs, const TensorOpCost &rhs) |
std::ostream & | operator<< (std::ostream &os, const TensorOpCost &tc) |
Definition at line 25 of file TensorCostModel.h.
|
inline |
Definition at line 55 of file TensorCostModel.h.
|
inline |
Definition at line 57 of file TensorCostModel.h.
|
inline |
Definition at line 63 of file TensorCostModel.h.
|
inlinestatic |
Definition at line 36 of file TensorCostModel.h.
|
inline |
Definition at line 74 of file TensorCostModel.h.
|
inline |
Definition at line 77 of file TensorCostModel.h.
|
inlinestatic |
Definition at line 49 of file TensorCostModel.h.
|
inline |
Definition at line 80 of file TensorCostModel.h.
|
inline |
Definition at line 106 of file TensorCostModel.h.
|
inline |
Definition at line 97 of file TensorCostModel.h.
|
inlinestatic |
Definition at line 40 of file TensorCostModel.h.
|
inline |
Definition at line 91 of file TensorCostModel.h.
|
inlinestatic |
Definition at line 45 of file TensorCostModel.h.
|
inlinestatic |
Definition at line 31 of file TensorCostModel.h.
|
inline |
Definition at line 122 of file TensorCostModel.h.
|
inline |
Definition at line 114 of file TensorCostModel.h.
|
inline |
Definition at line 83 of file TensorCostModel.h.
|
friend |
Definition at line 134 of file TensorCostModel.h.
|
friend |
Definition at line 139 of file TensorCostModel.h.
|
friend |
Definition at line 129 of file TensorCostModel.h.
|
friend |
Definition at line 145 of file TensorCostModel.h.
|
private |
Definition at line 152 of file TensorCostModel.h.
|
private |
Definition at line 153 of file TensorCostModel.h.
|
private |
Definition at line 154 of file TensorCostModel.h.