Offset utility for CPU. More...
#include <offset_util.hpp>
Inheritance diagram for lmp::tensor::detail::OffsetUtil:
Public Member Functions | |
| OffsetUtil (size_t ndim) | |
Public Attributes | |
| size_t | ndim |
Protected Member Functions | |
| std::vector< stride_t > | init_padded_strides (const std::vector< size_t > &shape, const std::vector< stride_t > &stride) const |
Detailed Description
Offset utility for CPU.
this class is used to get the offset of the elements in the tensor.
The documentation for this class was generated from the following files:
- include/lamppp/tensor/cpu/offset_util.hpp
- src/tensor/cpu/offset_util.cpp
