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.
- Parameters
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.
- Parameters
dims – [in] Define this operation will be performed on which axes.
-
std::vector<int64_t> get_dims() const
Method to get dims attribute.
- Returns
axes.
-
inline TileFusion()