diff options
author | Annie Tallund <annie.tallund@arm.com> | 2023-09-19 16:34:14 +0200 |
---|---|---|
committer | Annie Tallund <annie.tallund@arm.com> | 2023-09-28 13:04:15 +0200 |
commit | f692ee1dcfc02f338264221030a839f0c1d8d55c (patch) | |
tree | 29ab6e1b5d26e13d639171edfa5225d8e0433555 /src/mlia/target | |
parent | 3fb02429a868f5b9b2e4660bd33b8210bf54bee0 (diff) | |
download | mlia-f692ee1dcfc02f338264221030a839f0c1d8d55c.tar.gz |
MLIA-469 Support batch size > 1 for optimizations
- Add a PruningPolicy to skip layers that are not
supported by the Keras pruning API
- Make dataset generation more generic to support
use-cases beyond classification
Signed-off-by: Annie Tallund <annie.tallund@arm.com>
Change-Id: I198dae2b53860f449f2fdbc71575babceed1ffcf
Diffstat (limited to 'src/mlia/target')
0 files changed, 0 insertions, 0 deletions