Searched refs:TensorBlockResourceRequirements (Results 1 – 14 of 14) 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() argument137 static EIGEN_STRONG_INLINE TensorBlockResourceRequirements138 merge(const TensorBlockResourceRequirements& lhs, in merge() argument139 const TensorBlockResourceRequirements& rhs) { in merge()145 EIGEN_DEVICE_FUNC TensorBlockResourceRequirements& addCostPerCoeff( in addCostPerCoeff() argument[all …]
153 internal::TensorBlockResourceRequirements getResourceRequirements() const { in getResourceRequirements()154 return internal::TensorBlockResourceRequirements::any(); in getResourceRequirements()326 internal::TensorBlockResourceRequirements getResourceRequirements() const {327 return internal::TensorBlockResourceRequirements::any();522 internal::TensorBlockResourceRequirements getResourceRequirements() const {657 internal::TensorBlockResourceRequirements getResourceRequirements() const {659 return internal::TensorBlockResourceRequirements::merge(939 internal::TensorBlockResourceRequirements getResourceRequirements() const {944 internal::TensorBlockResourceRequirements::merge(then_req, else_req);948 return internal::TensorBlockResourceRequirements::merge(
204 internal::TensorBlockResourceRequirements getResourceRequirements() const {205 return internal::TensorBlockResourceRequirements::merge(
202 internal::TensorBlockResourceRequirements getResourceRequirements() const {203 return internal::TensorBlockResourceRequirements::any();
168 internal::TensorBlockResourceRequirements getResourceRequirements() const {171 return internal::TensorBlockResourceRequirements::skewed<Scalar>(
282 internal::TensorBlockResourceRequirements getResourceRequirements() const {284 return internal::TensorBlockResourceRequirements::merge(285 internal::TensorBlockResourceRequirements::skewed<Scalar>(target_size),
231 internal::TensorBlockResourceRequirements getResourceRequirements() const {233 return internal::TensorBlockResourceRequirements::merge(234 internal::TensorBlockResourceRequirements::skewed<Scalar>(target_size),
186 internal::TensorBlockResourceRequirements getResourceRequirements() const {187 return internal::TensorBlockResourceRequirements::any();608 internal::TensorBlockResourceRequirements getResourceRequirements() const {610 return internal::TensorBlockResourceRequirements::merge(611 internal::TensorBlockResourceRequirements::skewed<Scalar>(target_size),
232 internal::TensorBlockResourceRequirements getResourceRequirements() const {236 return internal::TensorBlockResourceRequirements::skewed<Scalar>(
234 internal::TensorBlockResourceRequirements getResourceRequirements() const {246 using BlockRequirements = internal::TensorBlockResourceRequirements;
168 internal::TensorBlockResourceRequirements getResourceRequirements() const {
619 internal::TensorBlockResourceRequirements getResourceRequirements() const {623 return internal::TensorBlockResourceRequirements::merge(625 internal::TensorBlockResourceRequirements::skewed<Scalar>(target_size));
193 const TensorBlockResourceRequirements requirements =247 TensorBlockResourceRequirements requirements =
428 internal::TensorBlockResourceRequirements getResourceRequirements() const {