Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::Tensor< TElementType, TMemoryResource > Member List

This is the complete list of members for Mila::Dnn::Tensor< TElementType, TMemoryResource >, including all inherited members.

allocateBuffer(TElementType value)Mila::Dnn::Tensor< TElementType, TMemoryResource >inlineprivate
at(const std::vector< size_t > &indices) constMila::Dnn::Tensor< TElementType, TMemoryResource >inline
buffer_Mila::Dnn::Tensor< TElementType, TMemoryResource >private
clone() constMila::Dnn::Tensor< TElementType, TMemoryResource >inline
computeIndex(const std::vector< size_t > &indices) constMila::Dnn::Tensor< TElementType, TMemoryResource >inlineprivate
computeSize(const std::vector< size_t > &shape)Mila::Dnn::Tensor< TElementType, TMemoryResource >inlineprivatestatic
computeStrides(const std::vector< size_t > &shape)Mila::Dnn::Tensor< TElementType, TMemoryResource >inlineprivatestatic
copyFrom(const Tensor< TElementType, SrcMemoryResource > &src)Mila::Dnn::Tensor< TElementType, TMemoryResource >inline
data()Mila::Dnn::Tensor< TElementType, TMemoryResource >inline
data() constMila::Dnn::Tensor< TElementType, TMemoryResource >inline
empty() constMila::Dnn::Tensor< TElementType, TMemoryResource >inline
external_memory_ptr_Mila::Dnn::Tensor< TElementType, TMemoryResource >private
fill(const TElementType &value)Mila::Dnn::Tensor< TElementType, TMemoryResource >inline
flatten()Mila::Dnn::Tensor< TElementType, TMemoryResource >inline
flattened() constMila::Dnn::Tensor< TElementType, TMemoryResource >inline
get_uid() constMila::Dnn::Tensor< TElementType, TMemoryResource >inline
getBufferString(size_t start_index=0, size_t depth=0) constMila::Dnn::Tensor< TElementType, TMemoryResource >inline
getName() constMila::Dnn::Tensor< TElementType, TMemoryResource >inline
is_device_accessible()Mila::Dnn::Tensor< TElementType, TMemoryResource >inlinestatic
is_host_accessible()Mila::Dnn::Tensor< TElementType, TMemoryResource >inlinestatic
MR typedefMila::Dnn::Tensor< TElementType, TMemoryResource >
name_Mila::Dnn::Tensor< TElementType, TMemoryResource >private
operator<<Mila::Dnn::Tensor< TElementType, TMemoryResource >friend
operator=(Tensor &&other) noexceptMila::Dnn::Tensor< TElementType, TMemoryResource >inline
operator=(const Tensor &other)Mila::Dnn::Tensor< TElementType, TMemoryResource >inline
operator[](Args... args)Mila::Dnn::Tensor< TElementType, TMemoryResource >inline
operator[](Args... args) constMila::Dnn::Tensor< TElementType, TMemoryResource >inline
outputBuffer(size_t index, size_t depth) constMila::Dnn::Tensor< TElementType, TMemoryResource >inlineprivate
rank() constMila::Dnn::Tensor< TElementType, TMemoryResource >inline
raw_data()Mila::Dnn::Tensor< TElementType, TMemoryResource >inline
raw_data() constMila::Dnn::Tensor< TElementType, TMemoryResource >inline
reshape(const std::vector< size_t > &new_shape)Mila::Dnn::Tensor< TElementType, TMemoryResource >inline
scalar_t typedefMila::Dnn::Tensor< TElementType, TMemoryResource >
set(const std::vector< size_t > &indices, TElementType value)Mila::Dnn::Tensor< TElementType, TMemoryResource >inline
set_uid()Mila::Dnn::Tensor< TElementType, TMemoryResource >inlineprivate
setName(const std::string &value)Mila::Dnn::Tensor< TElementType, TMemoryResource >inline
shape() constMila::Dnn::Tensor< TElementType, TMemoryResource >inline
shape_Mila::Dnn::Tensor< TElementType, TMemoryResource >private
size() constMila::Dnn::Tensor< TElementType, TMemoryResource >inline
size_Mila::Dnn::Tensor< TElementType, TMemoryResource >private
strides() constMila::Dnn::Tensor< TElementType, TMemoryResource >inline
strides_Mila::Dnn::Tensor< TElementType, TMemoryResource >private
Tensor(const std::vector< size_t > &shape, TElementType value=TElementType{})Mila::Dnn::Tensor< TElementType, TMemoryResource >inlineexplicit
Tensor(const std::vector< size_t > &shape, std::shared_ptr< TElementType > data_ptr)Mila::Dnn::Tensor< TElementType, TMemoryResource >inline
Tensor()Mila::Dnn::Tensor< TElementType, TMemoryResource >inline
Tensor(const Tensor &other)Mila::Dnn::Tensor< TElementType, TMemoryResource >inline
Tensor(Tensor &&other) noexceptMila::Dnn::Tensor< TElementType, TMemoryResource >inline
to() constMila::Dnn::Tensor< TElementType, TMemoryResource >inline
toFloat() constMila::Dnn::Tensor< TElementType, TMemoryResource >inline
toHalf() constMila::Dnn::Tensor< TElementType, TMemoryResource >inline
toHostAccessible() constMila::Dnn::Tensor< TElementType, TMemoryResource >inline
toString(bool showBuffer=false) constMila::Dnn::Tensor< TElementType, TMemoryResource >inline
uid_Mila::Dnn::Tensor< TElementType, TMemoryResource >private
validateIndices(const std::vector< size_t > &indices, const std::string &method_name) constMila::Dnn::Tensor< TElementType, TMemoryResource >inlineprivate
~Tensor()=defaultMila::Dnn::Tensor< TElementType, TMemoryResource >