Searched defs:value_cache (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/executorch/extension/llm/custom_ops/ |
| H A D | sdpa_with_kv_cache.py | 44 value_cache, argument
111 value_cache, argument
148 value_cache, argument
|
| H A D | op_sdpa_aot.cpp | 25 Tensor& value_cache, in sdpa_with_kv_cache_out_no_context()
58 at::Tensor& value_cache, in sdpa_with_kv_cache_aten()
|
| H A D | op_sdpa_with_kv_cache_test.cpp | 26 exec_aten::Tensor& value_cache, in op_sdpa_with_kv_cache()
|
| H A D | op_sdpa.cpp | 992 Tensor& value_cache, in sdpa_with_kv_cache_out()
|
| /aosp_15_r20/external/executorch/backends/vulkan/test/op_tests/ |
| H A D | sdpa_test.cpp | 37 Tensor& value_cache, in sdpa_with_kv_cache_out_no_context()
70 at::Tensor& value_cache, in sdpa_with_kv_cache_aten()
159 at::Tensor& value_cache, in sdpa_reference_impl()
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | profiler_python.cpp | 313 ValueCache* value_cache, in set_class()
916 const ValueCache& value_cache, in PostProcess()
943 const ValueCache& value_cache, in operator ()()
|
| /aosp_15_r20/external/mbedtls/scripts/mbedtls_dev/ |
| H A D | psa_storage.py | 35 value_cache = {} #type: Dict[str, int] variable in Expr
|
| /aosp_15_r20/external/openthread/third_party/mbedtls/repo/scripts/mbedtls_dev/ |
| H A D | psa_storage.py | 35 value_cache = {} #type: Dict[str, int] variable in Expr
|