allocateBuffer(TElementType value) | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inlineprivate |
at(const std::vector< size_t > &indices) const | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inline |
buffer_ | Mila::Dnn::Tensor< TElementType, TMemoryResource > | private |
clone() const | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inline |
computeIndex(const std::vector< size_t > &indices) const | Mila::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() const | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inline |
empty() const | Mila::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() const | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inline |
get_uid() const | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inline |
getBufferString(size_t start_index=0, size_t depth=0) const | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inline |
getName() const | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inline |
is_device_accessible() | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inlinestatic |
is_host_accessible() | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inlinestatic |
MR typedef | Mila::Dnn::Tensor< TElementType, TMemoryResource > | |
name_ | Mila::Dnn::Tensor< TElementType, TMemoryResource > | private |
operator<< | Mila::Dnn::Tensor< TElementType, TMemoryResource > | friend |
operator=(Tensor &&other) noexcept | Mila::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) const | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inline |
outputBuffer(size_t index, size_t depth) const | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inlineprivate |
rank() const | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inline |
raw_data() | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inline |
raw_data() const | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inline |
reshape(const std::vector< size_t > &new_shape) | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inline |
scalar_t typedef | Mila::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() const | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inline |
shape_ | Mila::Dnn::Tensor< TElementType, TMemoryResource > | private |
size() const | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inline |
size_ | Mila::Dnn::Tensor< TElementType, TMemoryResource > | private |
strides() const | Mila::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) noexcept | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inline |
to() const | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inline |
toFloat() const | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inline |
toHalf() const | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inline |
toHostAccessible() const | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inline |
toString(bool showBuffer=false) const | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inline |
uid_ | Mila::Dnn::Tensor< TElementType, TMemoryResource > | private |
validateIndices(const std::vector< size_t > &indices, const std::string &method_name) const | Mila::Dnn::Tensor< TElementType, TMemoryResource > | inlineprivate |
~Tensor()=default | Mila::Dnn::Tensor< TElementType, TMemoryResource > | |