| /external/executorch/examples/models/llava/runner/ |
| D | llava_runner.cpp | 100 int32_t seq_len, in generate_from_pos()
132 int32_t seq_len, in generate()
|
| /external/pytorch/benchmarks/fastrnns/ |
| 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
|
| /external/executorch/backends/vulkan/test/op_tests/ |
| 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
|
| D | rotary_embedding_test.cpp | 82 const int seq_len = 1) { in test_reference()
|
| /external/executorch/examples/models/llama/runner/ |
| D | runner.cpp | 153 int32_t seq_len, in generate()
279 Error Runner::warmup(const std::string& prompt, int32_t seq_len) { in warmup()
|
| /external/executorch/extension/llm/custom_ops/ |
| D | sdpa_with_kv_cache.py | 46 seq_len, argument
113 seq_len, argument
|
| 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()
|
| 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
|
| D | op_update_quantized_cache.cpp | 70 int64_t seq_len = value.size(1); in update_quantized_cache_out() local
|
| /external/executorch/examples/models/phi-3-mini/ |
| D | main.cpp | 43 int32_t seq_len = FLAGS_seq_len; in main() local
|
| D | runner.cpp | 62 std::size_t seq_len = input_tokens.size() + 1; in generate() local
|
| /external/executorch/examples/models/llama/ |
| D | main.cpp | 58 int32_t seq_len = FLAGS_seq_len; in main() local
|
| /external/executorch/examples/qualcomm/oss_scripts/llama2/ |
| D | qnn_llama_runner.cpp | 61 int32_t seq_len = FLAGS_seq_len; in main() local
|
| /external/executorch/examples/models/llava/ |
| D | main.cpp | 69 int32_t seq_len = FLAGS_seq_len; in main() local
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| 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
|
| D | ctc_decoder.h | 82 Status Decode(const typename CTCDecoder<T>::SequenceLength& seq_len, in Decode()
|
| 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()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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
|
| D | ctc_loss_op.cc | 93 const Tensor* seq_len; in Compute() local
268 const Tensor* seq_len; in Compute() local
|
| /external/executorch/examples/qualcomm/oss_scripts/llama2/runner/ |
| 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()
|
| /external/executorch/extension/llm/runner/ |
| D | text_token_generator.h | 49 int32_t seq_len, in generate()
|
| /external/tensorflow/tensorflow/lite/kernels/ctc/ |
| D | ctc_decoder.h | 77 bool Decode(const CTCDecoder::SequenceLength& seq_len, in Decode()
|
| /external/executorch/extension/android/jni/ |
| D | jni_layer_llama.cpp | 185 jint seq_len, in generate()
283 jint seq_len, in generate_from_pos()
|
| /external/executorch/examples/qualcomm/qaihub_scripts/llama/runner/ |
| 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
|
| /external/executorch/examples/qualcomm/oss_scripts/llama3_2/runner/ |
| D | runner.cpp | 120 int32_t seq_len, in generate()
|