| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_custom_lowering.py | 43 def j2pd_meta(inp, offsets, max_seq_len, pad_value): argument
50 def j2pd_cuda(inp, offsets, max_seq_len, pad_value): argument
62 def j2pd_lowering(inp, offsets, max_seq_len, pad_value): argument
136 def fn(inp, offsets, max_seq_len): argument
164 def fn(inp, offsets, max_seq_len): argument
|
| /aosp_15_r20/external/executorch/backends/vulkan/test/op_tests/ |
| H A D | sdpa_test.cpp | 137 const int max_seq_len = k_cache.size(1); in construct_attention_mask() local
220 const int max_seq_len, in test_reference_sdpa()
289 const int max_seq_len, in test_vulkan_sdpa()
447 const int max_seq_len = 7; in TEST() local
467 const int max_seq_len = 12; in TEST() local
486 const int max_seq_len = 128; in TEST() local
505 const int max_seq_len = 128; in TEST() local
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | SDPA.cpp | 154 const int64_t max_seq_len) { in add_cache_slice_view_node()
230 const int32_t max_seq_len = graph.size_at<int32_t>(1, k_cache); in sdpa_with_kv_cache_impl() local
|
| /aosp_15_r20/external/executorch/examples/models/phi-3-mini/ |
| H A D | runner.cpp | 52 void Runner::generate(const std::string& prompt, std::size_t max_seq_len) { in generate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | reverse_sequence_op.cc | 72 const int64_t max_seq_len = input_shape.dim_size(seq_dim_); in Compile() local
|
| /aosp_15_r20/external/executorch/examples/models/llava/ |
| H A D | model.py | 261 def __init__(self, use_sdpa_with_kv_cache_op=True, max_seq_len=768): argument
|
| /aosp_15_r20/external/executorch/extension/llm/export/ |
| H A D | builder.py | 69 max_seq_len, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | PackedSequence.cpp | 128 auto max_seq_len = batch_sizes_t.size(0); in _pack_padded_sequence_backward_symint() local
|
| /aosp_15_r20/external/executorch/extension/llm/custom_ops/ |
| H A D | test_sdpa_with_kv_cache.py | 408 max_seq_len, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ctc/ |
| H A D | ctc_loss_calculator.h | 152 auto max_seq_len = seq_len(0); in CalculateLoss() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/ |
| H A D | NestedTensorTransformerFunctions.cpp | 153 const auto max_seq_len = self.sizes()[2]; in NestedTensor_softmax_dropout() local
|
| /aosp_15_r20/external/executorch/examples/models/llama/ |
| H A D | llama_transformer.py | 92 max_seq_len: int = 2048 variable in ModelArgs
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/_tensor/ |
| H A D | common_dtensor.py | 95 max_seq_len: int = 16 variable in ModelArgs
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_nestedtensor.py | 297 def test_2d_nested_tensor(self, batch_size, max_seq_len, vocab_size): argument
318 def test_3d_nested_tensor(self, batch_size, max_seq_len, vocab_size): argument
340 def test_3d_nested_tensor_float(self, batch_size, max_seq_len, vocab_size): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | randomized_tests.cc | 3937 int max_seq_len = dims[seq_dim]; in TEST_F() local
|