Replace torch function with reshape_and_cache fused op for better performance. The `reshape_and_cache` function wasn't working because it expected torch.int32 tensor, but a torch.int64 tensor was provided. Signed-off-by: Jade Zheng <zheng.shoujian@outlook.com>