| 325 | } |
| 326 | |
| 327 | constexpr index_t __MATX_INLINE__ __MATX_HOST__ __MATX_DEVICE__ Size(int dim) const noexcept |
| 328 | { |
| 329 | if(dim==axis_) |
| 330 | return sizeof...(Ts); |
| 331 | else if (dim < axis_) { |
| 332 | return cuda::std::get<0>(ops_).Size(dim); |
| 333 | } else { |
| 334 | // remove axis_ dim from dim. |
| 335 | return cuda::std::get<0>(ops_).Size(dim-1); |
| 336 | } |
| 337 | } |
| 338 | |
| 339 | ~StackOp() = default; |
| 340 | StackOp(const StackOp &rhs) = default; |