#include <TensorBlock.h>
Public Member Functions | |
EIGEN_DEVICE_FUNC TensorBlockResourceRequirements & | addCostPerCoeff (TensorOpCost cost) |
Static Public Member Functions | |
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorBlockResourceRequirements | any () |
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorBlockResourceRequirements | merge (const TensorBlockResourceRequirements &lhs, const TensorBlockResourceRequirements &rhs) |
template<typename Scalar > | |
static EIGEN_DEVICE_FUNC TensorBlockResourceRequirements | skewed (size_t size_in_bytes) |
template<typename Scalar > | |
static EIGEN_DEVICE_FUNC TensorBlockResourceRequirements | uniform (size_t size_in_bytes) |
template<typename Scalar > | |
static EIGEN_DEVICE_FUNC TensorBlockResourceRequirements | withShapeAndSize (TensorBlockShapeType shape_type, size_t size_in_bytes) |
template<typename Scalar > | |
static EIGEN_DEVICE_FUNC TensorBlockResourceRequirements | withShapeAndSize (TensorBlockShapeType shape_type, size_t size_in_bytes, TensorOpCost cost) |
Public Attributes | |
TensorOpCost | cost_per_coeff |
TensorBlockShapeType | shape_type |
size_t | size |
Private Types | |
using | Requirements = TensorBlockResourceRequirements |
Static Private Member Functions | |
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE size_t | merge (size_t lhs_size, size_t rhs_size) |
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorBlockShapeType | merge (TensorBlockShapeType lhs, TensorBlockShapeType rhs) |
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost | merge (TensorOpCost lhs_cost, TensorOpCost rhs_cost) |
Definition at line 75 of file TensorBlock.h.
|
private |
Definition at line 160 of file TensorBlock.h.
|
inline |
Definition at line 145 of file TensorBlock.h.
|
inlinestatic |
Definition at line 155 of file TensorBlock.h.
|
inlinestatic |
Definition at line 138 of file TensorBlock.h.
|
inlinestaticprivate |
Definition at line 163 of file TensorBlock.h.
|
inlinestaticprivate |
Definition at line 169 of file TensorBlock.h.
|
inlinestaticprivate |
Definition at line 177 of file TensorBlock.h.
|
inlinestatic |
Definition at line 123 of file TensorBlock.h.
|
inlinestatic |
Definition at line 130 of file TensorBlock.h.
|
inlinestatic |
Definition at line 100 of file TensorBlock.h.
|
inlinestatic |
Definition at line 92 of file TensorBlock.h.
TensorOpCost Eigen::internal::TensorBlockResourceRequirements::cost_per_coeff |
Definition at line 78 of file TensorBlock.h.
TensorBlockShapeType Eigen::internal::TensorBlockResourceRequirements::shape_type |
Definition at line 76 of file TensorBlock.h.
size_t Eigen::internal::TensorBlockResourceRequirements::size |
Definition at line 77 of file TensorBlock.h.