Class TileFusion

Defined in File tile_fusion.h

Inheritance Relationships

Base Type

public mindspore::ops::Tile (Class Tile)

Class Documentation

class TileFusion : public mindspore::ops::Tile 

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.