Description

Traverses each key and value, and fills the key and value(num_heads, head_size) in the specified position of key_cache/value_cache with slotmapping.

Figure 1 ReshapeAndCacheOperation function diagram