| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | cudnn_rnn_ops_test.cc | 116 int max_seq_length = 2; in TEST() local
158 int max_seq_length = 2; in TEST() local
|
| H A D | cudnn_rnn_ops.cc | 194 auto max_seq_length = c->Dim(input_shape, 0); in __anonb0821bee0502() local
|
| /aosp_15_r20/external/pytorch/benchmarks/gpt_fast/ |
| H A D | model.py | 93 self, max_batch_size, max_seq_length, n_heads, head_dim, dtype=torch.bfloat16 argument
129 def setup_caches(self, max_batch_size, max_seq_length): argument
|
| H A D | mixtral_moe_model.py | 72 self, max_batch_size, max_seq_length, n_heads, head_dim, dtype=torch.bfloat16 argument
108 def setup_caches(self, max_batch_size, max_seq_length): argument
|
| /aosp_15_r20/external/executorch/examples/models/llama/source_transformation/ |
| H A D | quantized_kv_cache.py | 34 max_seq_length, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | PackedSequence.cpp | 149 int64_t max_seq_length = max_real_seq_length; in _pad_packed_sequence() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | stream_executor_pimpl.cc | 394 StreamExecutor::createRnnSequenceTensorDescriptor(int max_seq_length, in createRnnSequenceTensorDescriptor()
408 int max_seq_length, int batch_size, int data_size, in createRnnSequenceTensorDescriptor()
|
| H A D | dnn.h | 2060 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size, in createRnnSequenceTensorDescriptor()
2067 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size, in createRnnSequenceTensorDescriptor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 1635 CudnnRnnSequenceTensorDescriptor(GpuExecutor* parent, int max_seq_length, in CudnnRnnSequenceTensorDescriptor()
1653 GpuExecutor* parent, int max_seq_length, int batch_size, int data_size, in Create()
1671 GpuExecutor* parent, int max_seq_length, int batch_size, int data_size, in Create()
1710 int max_seq_length() const { return max_seq_length_; } in max_seq_length() function in stream_executor::gpu::CudnnRnnSequenceTensorDescriptor
1764 int max_seq_length = 0; member
2436 CudnnSupport::createRnnSequenceTensorDescriptor(int max_seq_length, in createRnnSequenceTensorDescriptor()
2449 int max_seq_length, int batch_size, int data_size, in createRnnSequenceTensorDescriptor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | ctc_ops.py | 556 def ctc_state_log_probs(seq_lengths, max_seq_length): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | cudnn_rnn_ops.cc | 154 int max_seq_length, int batch_size, int dir_count, in CudnnRnnParameters()
503 int max_seq_length; member
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_aot_inductor.py | 1442 max_seq_length, argument
|