| /aosp_15_r20/external/pytorch/benchmarks/fastrnns/ |
| H A D | custom_lstms.py | 403 def test_script_rnn_layer(seq_len, batch, input_size, hidden_size): argument
423 def test_script_stacked_rnn(seq_len, batch, input_size, hidden_size, num_layers): argument
449 def test_script_stacked_bidir_rnn(seq_len, batch, input_size, hidden_size, num_layers): argument
481 seq_len, batch, input_size, hidden_size, num_layers argument
494 def test_script_stacked_lnlstm(seq_len, batch, input_size, hidden_size, num_layers): argument
|
| /aosp_15_r20/external/executorch/backends/vulkan/test/op_tests/ |
| H A D | sdpa_test.cpp | 39 const int64_t seq_len, in sdpa_with_kv_cache_out_no_context()
72 const int64_t seq_len, in sdpa_with_kv_cache_aten()
138 const int seq_len = q.size(1); in construct_attention_mask() local
161 const int64_t seq_len, in sdpa_reference_impl()
385 int seq_len = base_sequence_len; in test_vulkan_sdpa() local
|
| /aosp_15_r20/external/executorch/examples/qualcomm/oss_scripts/llama2/runner/ |
| H A D | runner.cpp | 224 int32_t seq_len, in generate()
600 void IoMemMgr::set_all_shifted_ptrs(size_t seq_len) { in set_all_shifted_ptrs()
641 Error Runner::mem_alloc(size_t alignment, size_t seq_len) { in mem_alloc()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | ctc_decoder_ops.cc | 60 OpKernelContext* ctx, const Tensor** inputs, const Tensor** seq_len, in ValidateInputsGenerateOutputs()
195 const Tensor* seq_len; in Compute() local
300 const Tensor* seq_len; in Compute() local
|
| H A D | ctc_loss_op.cc | 93 const Tensor* seq_len; in Compute() local
268 const Tensor* seq_len; in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ctc/ |
| H A D | ctc_beam_search_test.cc | 157 Eigen::Map<const Eigen::ArrayXi> seq_len(&sequence_lengths[0], batch_size); in ctc_beam_search_decoding_with_and_without_dictionary() local
215 Eigen::Map<const Eigen::ArrayXi> seq_len(&sequence_lengths[0], batch_size); in ctc_beam_search_decoding_all_beam_elements_have_finite_scores() local
320 Eigen::Map<const Eigen::ArrayXi> seq_len(&sequence_lengths[0], batch_size); in ctc_beam_search_label_selection() local
|
| H A D | ctc_loss_calculator.h | 115 const VectorIn& seq_len, const LabelSequences& labels, in CalculateLoss()
289 int batch_size, int num_classes, const Vector& seq_len, in PopulateLPrimes()
|
| /aosp_15_r20/external/executorch/examples/models/llava/runner/ |
| H A D | llava_runner.cpp | 100 int32_t seq_len, in generate_from_pos()
132 int32_t seq_len, in generate()
|
| /aosp_15_r20/external/executorch/examples/models/llama/runner/ |
| H A D | runner.cpp | 153 int32_t seq_len, in generate()
279 Error Runner::warmup(const std::string& prompt, int32_t seq_len) { in warmup()
|
| /aosp_15_r20/external/executorch/extension/llm/custom_ops/ |
| H A D | sdpa_with_kv_cache.py | 46 seq_len, argument
113 seq_len, argument
|
| H A D | op_sdpa_aot.cpp | 27 const int64_t seq_len, in sdpa_with_kv_cache_out_no_context()
60 const int64_t seq_len, in sdpa_with_kv_cache_aten()
|
| H A D | test_sdpa_with_kv_cache.py | 17 def _sdpa_with_kv_cache_ref(q, k, v, k_cache, v_cache, attn_mask, start_pos, seq_len): argument
409 seq_len, argument
|
| H A D | op_update_quantized_cache.cpp | 70 int64_t seq_len = value.size(1); in update_quantized_cache_out() local
|
| /aosp_15_r20/packages/modules/Bluetooth/system/stack/hid/ |
| D | hidd_api.cc | 261 uint16_t seq_len = ((desc_len > 255) ? 5 : 4) + desc_len; in HID_DevAddRecord() local
311 uint8_t seq_len = 6; in HID_DevAddRecord() local
|
| /aosp_15_r20/external/executorch/extension/android/jni/ |
| H A D | jni_layer_llama.cpp | 185 jint seq_len, in generate()
283 jint seq_len, in generate_from_pos()
|
| /aosp_15_r20/packages/modules/Bluetooth/system/stack/sdp/ |
| D | sdp_discovery.cc | 524 uint32_t seq_len, attr_len; in save_attr_seq() local
590 uint32_t seq_len; in process_service_search_attr_rsp() local
|
| /aosp_15_r20/system/core/fs_mgr/libsnapshot/libsnapshot_cow/ |
| H A D | test_v3.cpp | 591 constexpr int seq_len = std::numeric_limits<uint16_t>::max() / sizeof(uint32_t) + 1; in TEST_F() local
638 const int seq_len = 10; in TEST_F() local
657 const int seq_len = 10; in TEST_F() local
|
| H A D | test_v2.cpp | 1254 const int seq_len = std::numeric_limits<uint16_t>::max() / sizeof(uint32_t) + 1; in TEST_F() local
1286 const int seq_len = 10; in TEST_F() local
1307 const int seq_len = 10; in TEST_F() local
|
| /aosp_15_r20/external/executorch/examples/qualcomm/qaihub_scripts/llama/runner/ |
| H A D | io_memory.cpp | 204 int seq_len = 1024, last_index = seq_len - 1; in update_io() local
414 int seq_len = 1023; in update_io() local
|
| /aosp_15_r20/external/executorch/examples/models/phi-3-mini/ |
| H A D | main.cpp | 43 int32_t seq_len = FLAGS_seq_len; in main() local
|
| H A D | runner.cpp | 62 std::size_t seq_len = input_tokens.size() + 1; in generate() local
|
| /aosp_15_r20/external/executorch/examples/models/llama/ |
| H A D | main.cpp | 58 int32_t seq_len = FLAGS_seq_len; in main() local
|
| /aosp_15_r20/external/executorch/examples/qualcomm/oss_scripts/llama2/ |
| H A D | qnn_llama_runner.cpp | 61 int32_t seq_len = FLAGS_seq_len; in main() local
|
| /aosp_15_r20/external/executorch/examples/models/llava/ |
| H A D | main.cpp | 69 int32_t seq_len = FLAGS_seq_len; in main() local
|
| /aosp_15_r20/external/executorch/extension/llm/runner/ |
| H A D | text_token_generator.h | 49 int32_t seq_len, in generate()
|