Returns a Tensor_view given location and size quantities
| 218 | |
| 219 | /// Returns a Tensor_view given location and size quantities |
| 220 | CUTLASS_HOST_DEVICE |
| 221 | TensorView subview( |
| 222 | TensorCoord extent, ///< extent of the resulting view |
| 223 | TensorCoord const& location = TensorCoord() ///< resulting view's origin within the old view |
| 224 | ) const { |
| 225 | |
| 226 | TensorView result(this->ref(), extent.clamp(extent_ - location)); |
| 227 | result.add_coord_offset(location); |
| 228 | return result; |
| 229 | } |
| 230 | |
| 231 | /// Returns the number of scalar elements needed to store tensor. |
| 232 | CUTLASS_HOST_DEVICE |