aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/nn/rewrite/core/train.py
diff options
context:
space:
mode:
authorNathan Bailey <nathan.bailey@arm.com>2024-02-27 12:46:52 +0000
committerNathan Bailey <nathan.bailey@arm.com>2024-04-16 12:57:10 +0100
commit427e02696f1ede596ef6dce82787a37e122efa78 (patch)
tree1fae7f7c8cb10af4f7c5119b73371b709c2c7caa /src/mlia/nn/rewrite/core/train.py
parent2973b6d52914023f9b8797aec8309957457d4189 (diff)
downloadmlia-427e02696f1ede596ef6dce82787a37e122efa78.tar.gz
feat: Implement the clustering rewrite for fp32
Implements a clustering rewrite for fully connected layers for fp32 models Resolves: MLIA-1079 Signed-off-by: Nathan Bailey <nathan.bailey@arm.com> Change-Id: I4c12f0bf911219b4066f0760976e424ebe900a0b
Diffstat (limited to 'src/mlia/nn/rewrite/core/train.py')
0 files changed, 0 insertions, 0 deletions