[bugfix] convert inp of index_select to enable cpu mode

[StrongSpoon]

PR Category

[OP Test]

Type of Change

[Buf Fix]

Description

convert input tensor and index to specific device for reference. otherwise the ref_out will not be on CPU when option --device is cpu.

Issue

Progress

• [x] Change is properly reviewed (1 reviewer required, 2 recommended).
• [x] Change is responded to an issue.
• [x] Change is fully covered by a UT.

Performance