Class TileFusion

Inheritance Relationships

Base Type

  • public mindspore::ops::BaseOperator

Class Documentation

class TileFusion : public mindspore::ops::BaseOperator

TileFusion defined Tile operator prototype of lite.

Public Functions

inline TileFusion()

Constructor.

void Init(const std::vector<int64_t> &dims)

Method to init the op's attributes.

参数

dims – [in] Define this operation will be performed on which axes.

void set_dims(const std::vector<int64_t> &dims)

Method to set dims attribute.

参数

dims – [in] Define this operation will be performed on which axes.

std::vector<int64_t> get_dims() const

Method to get dims attribute.

返回

axes.