fix strided slice if first dimension is not 1

xmos / ai_tools

AI applications and tools

Other

25 stars 10 forks source link

fix strided slice if first dimension is not 1 #876

Closed michaelpoluektov closed 9 months ago

michaelpoluektov commented 9 months ago

fix the replace slice pass in case the first dimension isn't 1.

This wasn't caught by CI because any sensible model has a batch size of 1, but caused an infinite loop when converting a pre-trained mobilenet_v1 (which included shape -> strided_slice ops later removed by the canonicalizer pass)