infini::ops::Embedding Member List

This is the complete list of members for infini::ops::Embedding, including all inherited members.

active_implementation_indices(Device::Type dev_type)	infini::ops::Operator< Embedding >	inlinestatic
active_implementation_indices(Device::Type dev_type)	infini::ops::Operator< Embedding >	inlinestatic
Call(const Handle &handle, const Config &config, const Args &... args)	infini::ops::Operator< Embedding >	static
Call(const Tensor tensor, const Args &... args)	infini::ops::Operator< Embedding >	inlinestatic
Call(const TensorLike &tensor, const Args &... args)	infini::ops::Operator< Embedding >	inlinestatic
Call(const Handle &handle, const Config &config, const Args &... args)	infini::ops::Operator< Embedding >	inlinestatic
Call(const Tensor tensor, const Args &... args)	infini::ops::Operator< Embedding >	inlinestatic
Call(const TensorLike &tensor, const Args &... args)	infini::ops::Operator< Embedding >	inlinestatic
clear_cache()	infini::ops::Operator< Embedding >	inlinestatic
clear_cache()	infini::ops::Operator< Embedding >	inlinestatic
config_ptr_	infini::ops::OperatorBase	protected
device_type_	infini::ops::Operator< Embedding >	protectedstatic
Embedding(const Tensor input, const Tensor weight, const std::optional< int64_t > padding_idx, const std::optional< double > max_norm, const double norm_type, const bool scale_grad_by_freq, const bool sparse, Tensor out)	infini::ops::Embedding	inline
Embedding(const Tensor input, const Tensor weight, Tensor out)	infini::ops::Embedding	inline
Embedding(const Tensor input, const Tensor weight, const int64_t padding_idx, const bool scale_grad_by_freq, const bool sparse, Tensor out)	infini::ops::Embedding	inline
embedding_dim_	infini::ops::Embedding	protected
handle_ptr_	infini::ops::OperatorBase	protected
implementation_index_	infini::ops::Operator< Embedding >	protectedstatic
input_dtype_	infini::ops::Embedding	protected
input_shape_	infini::ops::Embedding	protected
input_strides_	infini::ops::Embedding	protected
Make(const Config &config, const Tensor tensor, Args &&... args)	infini::ops::Operator< Embedding >	inlinestatic
Make(const Tensor tensor, Args &&... args)	infini::ops::Operator< Embedding >	inlinestatic
Make(const Config &config, const std::vector< Tensor > tensors, Args &&... args)	infini::ops::Operator< Embedding >	inlinestatic
Make(const std::vector< Tensor > tensors, Args &&... args)	infini::ops::Operator< Embedding >	inlinestatic
Make(const Config &config, const Tensor tensor, Args &&... args)	infini::ops::Operator< Embedding >	inlinestatic
Make(const Tensor tensor, Args &&... args)	infini::ops::Operator< Embedding >	inlinestatic
Make(const Config &config, const std::vector< Tensor > tensors, Args &&... args)	infini::ops::Operator< Embedding >	inlinestatic
Make(const std::vector< Tensor > tensors, Args &&... args)	infini::ops::Operator< Embedding >	inlinestatic
MakeReturnValue(const TensorLike &input, const TensorLike &weight, const std::optional< int64_t >=std::nullopt, const std::optional< double >=std::nullopt, const double=2.0, const bool=false, const bool=false)	infini::ops::Embedding	inlinestatic
max_norm_	infini::ops::Embedding	protected
norm_type_	infini::ops::Embedding	protected
num_indices_	infini::ops::Embedding	protected
NumIndices(const Tensor::Shape &input_shape)	infini::ops::Embedding	inlineprotectedstatic
operator()(const Tensor input, const Tensor weight, const std::optional< int64_t > padding_idx, const std::optional< double > max_norm, const double norm_type, const bool scale_grad_by_freq, const bool sparse, Tensor out) const =0	infini::ops::Embedding	pure virtual
operator()(const Tensor input, const Tensor weight, Tensor out) const	infini::ops::Embedding	inline
operator()(const Tensor input, const Tensor weight, const int64_t padding_idx, const bool scale_grad_by_freq, const bool sparse, Tensor out) const	infini::ops::Embedding	inline
Operator< Embedding >::operator()(const Handle &handle, const Args &... args)	infini::ops::Operator< Embedding >	inline
Operator< Embedding >::operator()(const Args &... args) const	infini::ops::Operator< Embedding >	inline
out_dtype_	infini::ops::Embedding	protected
out_shape_	infini::ops::Embedding	protected
out_strides_	infini::ops::Embedding	protected
padding_idx_	infini::ops::Embedding	protected
scale_grad_by_freq_	infini::ops::Embedding	protected
set_config(const Config &config)	infini::ops::OperatorBase	inline
set_config(const Config &config)	infini::ops::OperatorBase	inline
set_handle(const Handle &handle)	infini::ops::OperatorBase	inline
set_handle(const Handle &handle)	infini::ops::OperatorBase	inline
set_stream(void *stream)	infini::ops::OperatorBase	inline
set_stream(void *stream)	infini::ops::OperatorBase	inline
set_workspace(void *workspace)	infini::ops::OperatorBase	inline
set_workspace(void *workspace)	infini::ops::OperatorBase	inline
set_workspace_size_in_bytes(std::size_t workspace_size_in_bytes)	infini::ops::OperatorBase	inline
set_workspace_size_in_bytes(std::size_t workspace_size_in_bytes)	infini::ops::OperatorBase	inline
sparse_	infini::ops::Embedding	protected
stream_	infini::ops::OperatorBase	protected
vocab_size_	infini::ops::Embedding	protected
weight_dtype_	infini::ops::Embedding	protected
weight_shape_	infini::ops::Embedding	protected
weight_strides_	infini::ops::Embedding	protected
workspace_	infini::ops::OperatorBase	protected
workspace_size_in_bytes() const	infini::ops::OperatorBase	inlinevirtual
workspace_size_in_bytes() const	infini::ops::OperatorBase	inlinevirtual
workspace_size_in_bytes_	infini::ops::OperatorBase	protected
~OperatorBase()=default	infini::ops::OperatorBase	virtual
~OperatorBase()=default	infini::ops::OperatorBase	virtual