CUTLASS 3.5.1 (#1623)

* CUTLASS 3.5.1

* updates, optimizations, fixes

include/cutlass/layout/tensor.h

@@ -60,14 +60,19 @@ namespace layout {
 //
 /////////////////////////////////////////////////////////////////////////////////////////////////
 
-/// Tag used for 3-D NWC tensors for 1D conv, only used in 3.x API
+/// Tag used for 3-D NWC tensors for 1-D convolutions; only used in 3.x API
 class TensorNWC {};
 
-/// Tag used for n-D KCSRT tensors for nD conv, only used in 3.x API for wgrad output layouts
+/// Tag used for n-D KCSRT tensors for n-D convolutions; only used in 3.x API for wgrad output layouts
 class TensorKCS {};
 class TensorKCSR {};
 class TensorKCSRT {};
 
+/// Tag used for n-D CSRTK tensors for n-D convolutions; only used in 3.x API for wgrad output layouts
+class TensorCSK {};
+class TensorCSRK {};
+class TensorCSRTK {};
+
 /// Mapping function for 4-D NHWC tensors.
 class TensorNHWC {
 public:
@@ -639,14 +644,5 @@ public:
 
 /////////////////////////////////////////////////////////////////////////////////////////////////
 
-/// Tag used for linearized tensors with shape (NW, C) for 1D conv, only used in 3.x API
-class TensorLinearizedNWC {};
-/// Tag used for linearized tensors with shape (NHW, C) for 2D conv, only used in 3.x API
-class TensorLinearizedNHWC : public TensorNHWC {};
-/// Tag used for linearized tensors with shape (NDHW, C) for 3D conv, only used in 3.x API
-class TensorLinearizedNDHWC : public TensorNDHWC {};
-
-/////////////////////////////////////////////////////////////////////////////////////////////////
-
 } // namespace layout
 } // namespace cutlass