Commit Graph

2 Commits

Author SHA1 Message Date
He Wei 2d7c4334ec Support fusion_size argument for flatten_weights 2022-05-20 17:11:48 +08:00
He Wei ba0e450078 Add APIs to support flatten weights
```
Cell.flatten_weights()

Tensor._flatten_tensors()
Tensor._is_flattened()
Tensor._get_flattened_tensors()
```
2022-04-01 15:56:38 +08:00