ABACUS develop
Atomic-orbital Based Ab-initio Computation at UStc
Loading...
Searching...
No Matches
container::TensorMap Member List

This is the complete list of members for container::TensorMap, including all inherited members.

accessor() const &container::Tensorinline
accessor() &&=deletecontainer::Tensor
AllocateFrom(const Tensor &other, const TensorShape &shape)container::Tensor
buffer() constcontainer::Tensor
buffer_container::Tensorprotected
calculateLinearIndex(Indices... indices) constcontainer::Tensorinlineprotected
cast() constcontainer::Tensorinline
copy_from_device(const T *data, int64_t num_elements=-1)container::Tensorinline
CopyFrom(const Tensor &other)container::Tensor
CopyFrom(const Tensor &other, const TensorShape &shape)container::Tensor
CopyFromInternal(const Tensor &other, const TensorShape &shape)container::Tensorinlineprotected
data() constcontainer::Tensor
data() constcontainer::Tensorinline
data_type() constcontainer::Tensor
data_type_container::Tensorprotected
device_container::Tensorprotected
device_type() constcontainer::Tensor
get_value(Indices... indices) constcontainer::Tensorinline
GetAllocator(DeviceType device)container::Tensorstatic
inner_most_ptr(const int &index) constcontainer::Tensorinline
NumElements() constcontainer::Tensor
operator bool() constcontainer::Tensorinlineexplicit
operator=(const Tensor &other)container::Tensor
operator=(Tensor &&other) noexceptcontainer::Tensor
operator==(const Tensor &other) constcontainer::Tensor
operator[](const int &index) constcontainer::Tensor
reshape(TensorShape shape)container::Tensor
resize(const TensorShape &new_shape)container::Tensor
set_value(T value)container::Tensorinline
shape() constcontainer::Tensor
shape_container::Tensorprotected
shaped(const TensorShape &shape) constcontainer::Tensor
SizeOfType(DataType data_type)container::Tensorinlinestatic
slice(const std::vector< int > &start, const std::vector< int > &size) constcontainer::Tensor
sync(const Tensor &rhs)container::Tensor
Tensor()container::Tensor
Tensor(DataType data_type)container::Tensorexplicit
Tensor(DataType data_type, const TensorShape &shape)container::Tensor
Tensor(DataType data_type, DeviceType device, const TensorShape &shape)container::Tensor
Tensor(base::core::Allocator *a, DataType data_type, DeviceType device, const TensorShape &shape)container::Tensor
Tensor(const Tensor &other)container::Tensor
Tensor(Tensor &&other) noexceptcontainer::Tensor
Tensor(std::initializer_list< T > values, DeviceType device=DeviceType::CpuDevice)container::Tensorinline
TensorMap(void *data, DataType data_type, DeviceType device, const TensorShape &shape)container::TensorMap
TensorMap(void *data, const Tensor &other, const TensorShape &shape)container::TensorMap
TensorMap(void *data, const Tensor &other)container::TensorMap
to_device() constcontainer::Tensorinline
zero()container::Tensor
~Tensor()container::Tensor