infini::ops::Gemm Member List

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

a_strides_	infini::ops::Gemm	protected
a_type_	infini::ops::Gemm	protected
active_implementation_indices(Device::Type dev_type)	infini::ops::Operator< Gemm >	inlinestatic
active_implementation_indices(Device::Type dev_type)	infini::ops::Operator< Gemm >	inlinestatic
alpha_	infini::ops::Gemm	protected
b_strides_	infini::ops::Gemm	protected
b_type_	infini::ops::Gemm	protected
batch_count_	infini::ops::Gemm	protected
batch_stride_a_	infini::ops::Gemm	protected
batch_stride_b_	infini::ops::Gemm	protected
batch_stride_y_	infini::ops::Gemm	protected
beta_	infini::ops::Gemm	protected
Call(const Handle &handle, const Config &config, const Args &... args)	infini::ops::Operator< Gemm >	static
Call(const Tensor tensor, const Args &... args)	infini::ops::Operator< Gemm >	inlinestatic
Call(const TensorLike &tensor, const Args &... args)	infini::ops::Operator< Gemm >	inlinestatic
Call(const Handle &handle, const Config &config, const Args &... args)	infini::ops::Operator< Gemm >	inlinestatic
Call(const Tensor tensor, const Args &... args)	infini::ops::Operator< Gemm >	inlinestatic
Call(const TensorLike &tensor, const Args &... args)	infini::ops::Operator< Gemm >	inlinestatic
clear_cache()	infini::ops::Operator< Gemm >	inlinestatic
clear_cache()	infini::ops::Operator< Gemm >	inlinestatic
config_ptr_	infini::ops::OperatorBase	protected
device_type_	infini::ops::Operator< Gemm >	protectedstatic
EffectiveBeta(const std::optional< Tensor > &c, std::optional< float > beta)	infini::ops::Gemm	inlineprotectedstatic
Gemm(const Tensor a, const Tensor b, const std::optional< Tensor > c, std::optional< float > alpha, std::optional< float > beta, std::optional< int > trans_a, std::optional< int > trans_b, Tensor y)	infini::ops::Gemm	inline
Gemm(const Tensor a, const Tensor b, Tensor y)	infini::ops::Gemm	inline
handle_ptr_	infini::ops::OperatorBase	protected
implementation_index_	infini::ops::Operator< Gemm >	protectedstatic
k_	infini::ops::Gemm	protected
lda_	infini::ops::Gemm	protected
ldb_	infini::ops::Gemm	protected
ldy_	infini::ops::Gemm	protected
m_	infini::ops::Gemm	protected
Make(const Config &config, const Tensor tensor, Args &&... args)	infini::ops::Operator< Gemm >	inlinestatic
Make(const Tensor tensor, Args &&... args)	infini::ops::Operator< Gemm >	inlinestatic
Make(const Config &config, const std::vector< Tensor > tensors, Args &&... args)	infini::ops::Operator< Gemm >	inlinestatic
Make(const std::vector< Tensor > tensors, Args &&... args)	infini::ops::Operator< Gemm >	inlinestatic
Make(const Config &config, const Tensor tensor, Args &&... args)	infini::ops::Operator< Gemm >	inlinestatic
Make(const Tensor tensor, Args &&... args)	infini::ops::Operator< Gemm >	inlinestatic
Make(const Config &config, const std::vector< Tensor > tensors, Args &&... args)	infini::ops::Operator< Gemm >	inlinestatic
Make(const std::vector< Tensor > tensors, Args &&... args)	infini::ops::Operator< Gemm >	inlinestatic
MakeReturnValue(const TensorLike &a, const TensorLike &b)	infini::ops::Gemm	inlinestatic
n_	infini::ops::Gemm	protected
operator()(const Tensor a, const Tensor b, const std::optional< Tensor > c, std::optional< float > alpha, std::optional< float > beta, std::optional< int > trans_a, std::optional< int > trans_b, Tensor y) const =0	infini::ops::Gemm	pure virtual
operator()(const Tensor a, const Tensor b, Tensor y) const	infini::ops::Gemm	inlinevirtual
Operator< Gemm >::operator()(const Handle &handle, const Args &... args)	infini::ops::Operator< Gemm >	inline
Operator< Gemm >::operator()(const Args &... args) const	infini::ops::Operator< Gemm >	inline
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
stream_	infini::ops::OperatorBase	protected
trans_a_	infini::ops::Gemm	protected
trans_b_	infini::ops::Gemm	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
y_strides_	infini::ops::Gemm	protected
y_type_	infini::ops::Gemm	protected
~OperatorBase()=default	infini::ops::OperatorBase	virtual
~OperatorBase()=default	infini::ops::OperatorBase	virtual