Qualcomm AI Engine Direct - increase index_put coverage

- refactor a bit & add more test cases

Author: haowhsu-quic