Searched defs:TensorBlockResourceRequirements (Results 1 – 1 of 1) sorted by relevance
75 struct TensorBlockResourceRequirements { struct85 TensorBlockResourceRequirements(TensorBlockShapeType shape_type_, size_t size_, in TensorBlockResourceRequirements() argument92 EIGEN_DEVICE_FUNC static TensorBlockResourceRequirements withShapeAndSize( in withShapeAndSize() argument100 EIGEN_DEVICE_FUNC static TensorBlockResourceRequirements withShapeAndSize( in withShapeAndSize() argument123 EIGEN_DEVICE_FUNC static TensorBlockResourceRequirements skewed( in skewed() argument130 EIGEN_DEVICE_FUNC static TensorBlockResourceRequirements uniform( in uniform() argument138 merge(const TensorBlockResourceRequirements& lhs, in merge() argument145 EIGEN_DEVICE_FUNC TensorBlockResourceRequirements& addCostPerCoeff( in addCostPerCoeff() argument155 static EIGEN_STRONG_INLINE TensorBlockResourceRequirements any() { in any() argument163 static EIGEN_STRONG_INLINE size_t merge(size_t lhs_size, size_t rhs_size) { in merge()[all …]