lmp::tensor::detail::cpu::CPUOffsetUtil< NArgs > Class Template Reference

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:

• csrc/include/lamppp/tensor/cpu/offset_util.hpp
• csrc/src/tensor/cpu/offset_util.cpp