| /external/executorch/extension/llm/custom_ops/ |
| D | sdpa_with_kv_cache.py | 44 value_cache, argument
74 value_cache.dim() == 4
81 value_cache.dtype == torch.float32
85 key_cache.size() == value_cache.size()
111 value_cache, argument
124 value_cache,
148 value_cache, argument
159 value_cache,
161 value_cache,
|
| D | op_sdpa_aot.cpp | 25 Tensor& value_cache, in sdpa_with_kv_cache_out_no_context() argument
43 value_cache, in sdpa_with_kv_cache_out_no_context()
58 at::Tensor& value_cache, in sdpa_with_kv_cache_aten() argument
74 value_cache, in sdpa_with_kv_cache_aten()
|
| D | op_sdpa.h | 24 Tensor& value_cache,
|
| D | op_sdpa.cpp | 992 Tensor& value_cache, in sdpa_with_kv_cache_out() argument
1004 validate_cache_params(key_cache, value_cache, start_pos, seq_len), in sdpa_with_kv_cache_out()
1011 update_cache(v_projected, value_cache, start_pos, seq_len); in sdpa_with_kv_cache_out()
1017 value_cache, in sdpa_with_kv_cache_out()
|
| D | op_sdpa_with_kv_cache_test.cpp | 26 exec_aten::Tensor& value_cache, in op_sdpa_with_kv_cache() argument
41 value_cache, in op_sdpa_with_kv_cache()
|
| /external/openthread/third_party/mbedtls/repo/scripts/mbedtls_dev/ |
| D | psa_storage.py | 35 value_cache = {} #type: Dict[str, int] variable in Expr
55 self.value_cache[e] = int(v, 0)
74 if normalized not in self.value_cache:
76 self.value_if_known = self.value_cache[normalized]
|
| /external/mbedtls/scripts/mbedtls_dev/ |
| D | psa_storage.py | 35 value_cache = {} #type: Dict[str, int] variable in Expr
55 self.value_cache[e] = int(v, 0)
74 if normalized not in self.value_cache:
76 self.value_if_known = self.value_cache[normalized]
|
| /external/executorch/backends/vulkan/test/op_tests/ |
| D | sdpa_test.cpp | 37 Tensor& value_cache, in sdpa_with_kv_cache_out_no_context() argument
55 value_cache, in sdpa_with_kv_cache_out_no_context()
70 at::Tensor& value_cache, in sdpa_with_kv_cache_aten() argument
86 value_cache, in sdpa_with_kv_cache_aten()
159 at::Tensor& value_cache, in sdpa_reference_impl() argument
173 value_cache, v_projected, 1, start_pos, start_pos + v_projected.size(1)); in sdpa_reference_impl()
177 value_cache = value_cache_updated; in sdpa_reference_impl()
|
| /external/pytorch/torch/csrc/autograd/ |
| D | profiler_python.cpp | 313 ValueCache* value_cache, in set_class() argument
321 value_cache->store<CallType::PyCall>(*cache.location_, no_ephemeral_t()); in set_class()
524 ValueCache& value_cache) { in intern()
527 value_cache.store<C>(callsite.value_, ephemeral); in intern()
528 value_cache.store<CallType::PyCall>(callsite.caller_, no_ephemeral_t()); in intern()
534 auto lookup(Callsite<C>& callsite, ValueCache& value_cache) const { in lookup()
536 value_cache.load<C>(callsite.value_), in lookup()
537 value_cache.load<CallType::PyCall>(callsite.caller_)); in lookup()
628 ValueCache* value_cache, in ThreadLocalResults()
632 value_cache_{value_cache},
[all …]
|