Hello, I'm working on a speech classification model built on Pytorch's nn.Conv1d. PyTorch expects the conv1d input shape to be (batch, channel, length), in my case (time_stamps, 1, feat_dimension).
After converting to TFLite, MLIA analysis shows that Conv ops only run on the NPU if batch_size=1.
If batch_size>1, it runs on the CPU with the message: "IFM Tensor batch size must be 1."
Since my user case requires time_stamps > 1 (a.k.a batch_size > 1), how can I solve to ensure Conv ops run on NPU?
Thank you,Hao