Searched defs:key_cache_0 (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/executorch/extension/llm/custom_ops/ |
| H A D | op_sdpa_with_kv_cache_test.cpp | 139 exec_aten::Tensor key_cache_0 = tfFloat.zeros({1, 5, 4, 4}); in TEST() local
381 exec_aten::Tensor key_cache_0 = tfFloat.zeros({1, 8, 7, 4}); in TEST() local
585 exec_aten::Tensor key_cache_0 = tfFloat.zeros({1, 5, 4, 4}); in TEST() local
832 exec_aten::Tensor key_cache_0 = tfFloat.zeros({1, 5, 8, 4}); in TEST() local
|