Searched defs:value_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 | 140 exec_aten::Tensor value_cache_0 = tfFloat.zeros({1, 5, 4, 4}); in TEST() local
382 exec_aten::Tensor value_cache_0 = tfFloat.zeros({1, 8, 7, 4}); in TEST() local
586 exec_aten::Tensor value_cache_0 = tfFloat.zeros({1, 5, 4, 4}); in TEST() local
833 exec_aten::Tensor value_cache_0 = tfFloat.zeros({1, 5, 8, 4}); in TEST() local
|