Inheritance diagram for lmp::tensor::detail::cpu::CPUOffsetUtil< NArgs >:
Public Member Functions | |
| CPUOffsetUtil (::std::array< const TensorImpl *, NArgs > ins, const TensorImpl &outs) | |
| ::std::array< stride_t, NArgs+1 > | get (size_t idx) const |
 Public Member Functions inherited from lmp::tensor::detail::OffsetUtil | |
| OffsetUtil (size_t ndim) | |
Public Attributes | |
| ::std::array< std::vector< stride_t >, NArgs+1 > | arg_strides_ |
| Public Attributes inherited from lmp::tensor::detail::OffsetUtil | |
| size_t | ndim |
Additional Inherited Members | |
| Protected Member Functions inherited from lmp::tensor::detail::OffsetUtil | |
| std::vector< stride_t > | init_padded_strides (const std::vector< size_t > &shape, const std::vector< stride_t > &stride) const |
The documentation for this class was generated from the following files:
- include/lamppp/tensor/cpu/offset_util.hpp
- src/tensor/cpu/offset_util.cpp
