Class Adder

Inheritance Relationships

Base Type

Derived Type

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.