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 slot mapping.

Figure 1 ReshapeAndCacheOperation function diagram