Searched defs:k_val (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/executorch/backends/apple/coreml/test/ |
| H A D | test_coreml_partitioner.py | 104 def forward(self, q, k_val, input_pos): argument
|
| /aosp_15_r20/external/executorch/examples/models/llama/source_transformation/ |
| H A D | quantized_kv_cache.py | 98 def update(self, input_pos, k_val, v_val): argument
|
| /aosp_15_r20/external/pytorch/benchmarks/gpt_fast/ |
| H A D | model.py | 100 def update(self, input_pos, k_val, v_val): argument
|
| H A D | mixtral_moe_model.py | 79 def update(self, input_pos, k_val, v_val): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lhlo_gpu_to_tfrt_gpu/ |
| H A D | gemm_pattern.cc | 137 auto k_val = lhs_matrix.transpose ? lhs_matrix.num_rows : lhs_matrix.num_cols; in CreateTfrtOps() local
|
| /aosp_15_r20/external/executorch/exir/backend/test/ |
| H A D | test_partitioner.py | 674 def forward(self, q, k_val, input_pos): argument
|
| /aosp_15_r20/external/coreboot/src/drivers/analogix/anx7625/ |
| H A D | anx7625.h | 353 unsigned int k_val; member
|
| /aosp_15_r20/external/executorch/backends/qualcomm/tests/ |
| H A D | models.py | 567 def forward(self, input_pos, k_val): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | batch_norm_kernel.cpp | 1265 opmath_t k_val = dotp_val * invstd_val * invstd_val / N; in batch_norm_cpu_backward_channels_last_internal() local
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_aot_inductor.py | 1452 def update(self, input_pos, k_val, v_val): argument
|