Correctly pass number of input allocations for BLAS am: 166b402861
am: 72f45777d5

Change-Id: I7b37176fb85180cf6576daef8edf34f93cc1b1a0