Class Adder
Defined in File adder.h
Inheritance Relationships
Base Type
public mindspore::ops::PrimitiveC
(Class PrimitiveC)
Derived Type
public mindspore::ops::AdderFusion
(Class AdderFusion)
Class Documentation
-
class Adder : public mindspore::ops::PrimitiveC
All defined All operator prototype of lite.
Subclassed by mindspore::ops::AdderFusion
Public Functions
-
inline explicit Adder(const std::string &k_name = kNameAdder)
Constructor.
-
~Adder() = default
Destructor.
-
void Init(const int64_t in_channel, const int64_t out_channel, const std::vector<int64_t> &kernel_size, const PadMode &pad_mode, const std::vector<int64_t> &stride, const std::vector<int64_t> &pad_list, const std::vector<int64_t> &dilation, const int64_t group, const Format &format)
Method to init the op’s attributes.
- Parameters
in_channel – [in] Define the input channel.
out_channel – [in] Define the output channel.
kernel_size – [in] Define the kernel size.
pad_mode – [in] Define the pad mode.
stride – [in] Define the stride.
pad_list – [in] Define the pad list.
dilation – [in] Define the dilation.
group – [in] Define the group.
format – [in] Define the format.
-
void set_in_channel(const int64_t in_channel)
Method to set in_channel attributes.
- Parameters
in_channel – [in] Define the input channel.
-
void set_out_channel(const int64_t out_channel)
Method to set out_channel attributes.
- Parameters
out_channel – [in] Define the output channel.
-
void set_kernel_size(const std::vector<int64_t> &kernel_size)
Method to set kernel_size attributes.
- Parameters
kernel_size – [in] Define the kernel size.
-
void set_pad_mode(const PadMode &pad_mode)
Method to set pad_mode attributes.
- Parameters
pad_mode – [in] Define the pad mode.
-
void set_stride(const std::vector<int64_t> &stride)
Method to set stride attributes.
- Parameters
stride – [in] Define the stride.
-
void set_pad_list(const std::vector<int64_t> &pad_list)
Method to set pad_list attributes.
- Parameters
pad_list – [in] Define the pad list.
-
void set_dilation(const std::vector<int64_t> &dilation)
Method to set dilation attributes.
- Parameters
dilation – [in] Define the dilation.
-
void set_group(const int64_t group)
Method to set group attributes.
- Parameters
group – [in] Define the group.
-
void set_format(const Format &format)
Method to set format attributes.
- Parameters
format – [in] Define the format.
-
int64_t get_in_channel() const
Method to get in_channel attributes.
- Returns
in_channel attributes.
-
int64_t get_out_channel() const
Method to get out_channel attributes.
- Returns
out_channel attributes.
-
std::vector<int64_t> get_kernel_size() const
Method to get kernel_size attributes.
- Returns
kernel_size attributes.
-
PadMode get_pad_mode() const
Method to get pad_mode attributes.
- Returns
pad_mode attributes.
-
std::vector<int64_t> get_stride() const
Method to get stride attributes.
- Returns
stride attributes.
-
std::vector<int64_t> get_pad_list() const
Method to get pad_list attributes.
- Returns
pad_list attributes.
-
std::vector<int64_t> get_dilation() const
Method to get dilation attributes.
- Returns
dilation attributes.
-
int64_t get_group() const
Method to get group attributes.
- Returns
group attributes.
-
Format get_format() const
Method to get format attributes.
- Returns
format attributes.
-
inline explicit Adder(const std::string &k_name = kNameAdder)