Class SplitWithOverlap

Inheritance Relationships

Base Type

  • public mindspore::ops::BaseOperator

Class Documentation

class SplitWithOverlap : public mindspore::ops::BaseOperator

All defined All operator prototype of lite.

Public Functions

inline SplitWithOverlap()

Constructor.

void Init(int64_t number_split, const std::vector<int64_t> &ratio, const std::vector<int64_t> &extend_top, const std::vector<int64_t> &extend_bottom, int64_t split_dim, int64_t stride, int64_t pad_top, bool trans_format)

Method to init the op’s attributes.

参数

  • number_split – [in] Define the number split.

  • ratio – [in] Define the ratio.

  • extend_top – [in] Define the extend top.

  • extend_bottom – [in] Define the extend bottom.

  • split_dim – [in] Define the split dim.

  • stride – [in] Define the pad stride.

  • pad_top – [in] Define the pad top.

  • trans_format – [in] Define the trans format.

void set_ratio(const std::vector<int64_t> &ratio)

Method to set ratio attributes.

参数

ratio – [in] Define the ratio.

void set_extend_top(const std::vector<int64_t> &extend_top)

Method to set extend_top attributes.

参数

extend_top – [in] Define the extend top.

void set_extend_bottom(const std::vector<int64_t> &extend_bottom)

Method to set extend_bottom attributes.

参数

extend_bottom – [in] Define the extend bottom.

void set_number_split(int64_t number_split)

Method to set number_split attributes.

参数

number_split – [in] Define the number split.

void set_split_dim(int64_t split_dim)

Method to set split_dim attributes.

参数

split_dim – [in] Define the split dim.

void set_split_stride(int64_t stride)

Method to set stride attributes.

参数

stride – [in] Define the stride.

void set_pad_top(int64_t pad_top)

Method to set pad_top attributes.

参数

pad_top – [in] Define the pad top.

void set_trans_format(bool trans_format)

Method to set trans_format attributes.

参数

trans_format – [in] Define the trans format.

std::vector<int64_t> get_ratio() const

Method to get ratio attributes.

返回

ratio attributes.

std::vector<int64_t> get_extend_top() const

Method to get extend_top attributes.

返回

extend_top attributes.

std::vector<int64_t> get_extend_bottom() const

Method to get extend_bottom attributes.

返回

extend_bottom attributes.

int64_t get_number_split() const

Method to get number_split attributes.

返回

number_split attributes.

int64_t get_split_dim() const

Method to get split_dim attributes.

返回

split_dim attributes.

int64_t get_split_stride() const

Method to get split_stride attributes.

返回

split_stride attributes.

int64_t get_pad_top() const

Method to get pad_top attributes.

返回

pad_top attributes.

bool get_trans_format() const

Method to get trans_format attributes.

返回

trans_format attributes.