Searched refs:withShapeAndSize (Results 1 – 1 of 1) sorted by relevance
92 EIGEN_DEVICE_FUNC static TensorBlockResourceRequirements withShapeAndSize( in withShapeAndSize() function100 EIGEN_DEVICE_FUNC static TensorBlockResourceRequirements withShapeAndSize( in withShapeAndSize() function116 return withShapeAndSize<Scalar>(shape_type, size_in_bytes, in withShapeAndSize()125 return withShapeAndSize<Scalar>(TensorBlockShapeType::kSkewedInnerDims, in skewed()132 return withShapeAndSize<Scalar>(TensorBlockShapeType::kUniformAllDims, in uniform()