LBANN  0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
lbann::cuTENSOR_PermuteImpl Member List

This is the complete list of members for lbann::cuTENSOR_PermuteImpl, including all inherited members.

cuTENSOR_PermuteImpl(ColMajorPerm perm)lbann::cuTENSOR_PermuteImplinline
DimsType typedeflbann::cuTENSOR_PermuteImpl
get_desc_key(El::Matrix< DataT, El::Device::GPU > const &mat, DimsType const &dims)lbann::cuTENSOR_PermuteImplprivatestatic
get_descriptor(El::Matrix< DataT, El::Device::GPU > const &mat, DimsType const &dims)lbann::cuTENSOR_PermuteImplprivatestatic
input_dims() const noexceptlbann::cuTENSOR_PermuteImplinline
input_modes() const noexceptlbann::cuTENSOR_PermuteImplinline
input_strides() constlbann::cuTENSOR_PermuteImplinline
inverse_permute(El::Matrix< DataT, El::Device::GPU > const &in, El::Matrix< DataT, El::Device::GPU > &out) constlbann::cuTENSOR_PermuteImpl
m_desc_maplbann::cuTENSOR_PermuteImplinlineprivatestatic
m_input_dimslbann::cuTENSOR_PermuteImplprivate
m_input_modeslbann::cuTENSOR_PermuteImplprivate
m_output_dimslbann::cuTENSOR_PermuteImplprivate
m_output_modeslbann::cuTENSOR_PermuteImplprivate
m_permlbann::cuTENSOR_PermuteImplprivate
make_modes(size_t const ndims)lbann::cuTENSOR_PermuteImplinlineprivatestatic
ModesType typedeflbann::cuTENSOR_PermuteImpl
output_dims() const noexceptlbann::cuTENSOR_PermuteImplinline
output_modes() const noexceptlbann::cuTENSOR_PermuteImplinline
output_strides() constlbann::cuTENSOR_PermuteImplinline
perm() const noexceptlbann::cuTENSOR_PermuteImplinline
permute(El::Matrix< DataT, El::Device::GPU > const &in, El::Matrix< DataT, El::Device::GPU > &out) constlbann::cuTENSOR_PermuteImpl
set_dims(DimsType input_dims)lbann::cuTENSOR_PermuteImplinline
StridesType typedeflbann::cuTENSOR_PermuteImpl
swap(cuTENSOR_PermuteImpl &other)lbann::cuTENSOR_PermuteImplinline