infini::ops::internal::BatchNormWithUpdate Member List

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

active_implementation_indices(Device::Type dev_type)	infini::ops::Operator< BatchNormWithUpdate >	inlinestatic
active_implementation_indices(Device::Type dev_type)	infini::ops::Operator< BatchNormWithUpdate >	inlinestatic
BatchNormWithUpdate(const Tensor input, const std::optional< Tensor > weight, const std::optional< Tensor > bias, Tensor running_mean, Tensor running_var, const double momentum, const double eps, Tensor out, Tensor save_mean, Tensor save_invstd, Tensor reserve)	infini::ops::internal::BatchNormWithUpdate	inline
bias_shape_	infini::ops::internal::BatchNormWithUpdate	protected
bias_strides_	infini::ops::internal::BatchNormWithUpdate	protected
bias_type_	infini::ops::internal::BatchNormWithUpdate	protected
Call(const Handle &handle, const Config &config, const Args &... args)	infini::ops::Operator< BatchNormWithUpdate >	static
Call(const Tensor tensor, const Args &... args)	infini::ops::Operator< BatchNormWithUpdate >	inlinestatic
Call(const TensorLike &tensor, const Args &... args)	infini::ops::Operator< BatchNormWithUpdate >	inlinestatic
Call(const Handle &handle, const Config &config, const Args &... args)	infini::ops::Operator< BatchNormWithUpdate >	inlinestatic
Call(const Tensor tensor, const Args &... args)	infini::ops::Operator< BatchNormWithUpdate >	inlinestatic
Call(const TensorLike &tensor, const Args &... args)	infini::ops::Operator< BatchNormWithUpdate >	inlinestatic
clear_cache()	infini::ops::Operator< BatchNormWithUpdate >	inlinestatic
clear_cache()	infini::ops::Operator< BatchNormWithUpdate >	inlinestatic
config_ptr_	infini::ops::OperatorBase	protected
device_index_	infini::ops::internal::BatchNormWithUpdate	protected
device_type_	infini::ops::Operator< BatchNormWithUpdate >	protectedstatic
eps_	infini::ops::internal::BatchNormWithUpdate	protected
handle_ptr_	infini::ops::OperatorBase	protected
has_bias_	infini::ops::internal::BatchNormWithUpdate	protected
has_weight_	infini::ops::internal::BatchNormWithUpdate	protected
implementation_index_	infini::ops::Operator< BatchNormWithUpdate >	protectedstatic
input_shape_	infini::ops::internal::BatchNormWithUpdate	protected
input_strides_	infini::ops::internal::BatchNormWithUpdate	protected
input_type_	infini::ops::internal::BatchNormWithUpdate	protected
Make(const Config &config, const Tensor tensor, Args &&... args)	infini::ops::Operator< BatchNormWithUpdate >	inlinestatic
Make(const Tensor tensor, Args &&... args)	infini::ops::Operator< BatchNormWithUpdate >	inlinestatic
Make(const Config &config, const std::vector< Tensor > tensors, Args &&... args)	infini::ops::Operator< BatchNormWithUpdate >	inlinestatic
Make(const std::vector< Tensor > tensors, Args &&... args)	infini::ops::Operator< BatchNormWithUpdate >	inlinestatic
Make(const Config &config, const Tensor tensor, Args &&... args)	infini::ops::Operator< BatchNormWithUpdate >	inlinestatic
Make(const Tensor tensor, Args &&... args)	infini::ops::Operator< BatchNormWithUpdate >	inlinestatic
Make(const Config &config, const std::vector< Tensor > tensors, Args &&... args)	infini::ops::Operator< BatchNormWithUpdate >	inlinestatic
Make(const std::vector< Tensor > tensors, Args &&... args)	infini::ops::Operator< BatchNormWithUpdate >	inlinestatic
momentum_	infini::ops::internal::BatchNormWithUpdate	protected
operator()(const Tensor input, const std::optional< Tensor > weight, const std::optional< Tensor > bias, Tensor running_mean, Tensor running_var, const double momentum, const double eps, Tensor out, Tensor save_mean, Tensor save_invstd, Tensor reserve) const =0	infini::ops::internal::BatchNormWithUpdate	pure virtual
Operator< BatchNormWithUpdate >::operator()(const Handle &handle, const Args &... args)	infini::ops::Operator< BatchNormWithUpdate >	inline
Operator< BatchNormWithUpdate >::operator()(const Args &... args) const	infini::ops::Operator< BatchNormWithUpdate >	inline
out_shape_	infini::ops::internal::BatchNormWithUpdate	protected
out_strides_	infini::ops::internal::BatchNormWithUpdate	protected
out_type_	infini::ops::internal::BatchNormWithUpdate	protected
reserve_shape_	infini::ops::internal::BatchNormWithUpdate	protected
reserve_strides_	infini::ops::internal::BatchNormWithUpdate	protected
reserve_type_	infini::ops::internal::BatchNormWithUpdate	protected
running_mean_shape_	infini::ops::internal::BatchNormWithUpdate	protected
running_mean_strides_	infini::ops::internal::BatchNormWithUpdate	protected
running_mean_type_	infini::ops::internal::BatchNormWithUpdate	protected
running_var_shape_	infini::ops::internal::BatchNormWithUpdate	protected
running_var_strides_	infini::ops::internal::BatchNormWithUpdate	protected
running_var_type_	infini::ops::internal::BatchNormWithUpdate	protected
save_invstd_shape_	infini::ops::internal::BatchNormWithUpdate	protected
save_invstd_strides_	infini::ops::internal::BatchNormWithUpdate	protected
save_invstd_type_	infini::ops::internal::BatchNormWithUpdate	protected
save_mean_shape_	infini::ops::internal::BatchNormWithUpdate	protected
save_mean_strides_	infini::ops::internal::BatchNormWithUpdate	protected
save_mean_type_	infini::ops::internal::BatchNormWithUpdate	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
stream_	infini::ops::OperatorBase	protected
weight_shape_	infini::ops::internal::BatchNormWithUpdate	protected
weight_strides_	infini::ops::internal::BatchNormWithUpdate	protected
weight_type_	infini::ops::internal::BatchNormWithUpdate	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