| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | lstm.py | 122 def RandomInputs(batch_size, seq_length, num_inputs): argument
135 def BuildLSTMLayer(batch_size, seq_length, num_inputs, num_nodes): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | cudnn_rnn_ops_test.cc | 42 int seq_length = 2; in TEST() local
79 int seq_length = 2; in TEST() local
|
| H A D | cudnn_rnn_ops.cc | 91 auto seq_length = c->Dim(input_shape, 0); in __anonb0821bee0302() local
140 auto seq_length = c->Dim(input_shape, 0); in __anonb0821bee0402() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Gru.cpp | 70 const auto seq_length = input_vk.size(1); in gru_input() local
295 const auto seq_length = input_vk.size(1); in run_gru_context() local
|
| H A D | Lstm.cpp | 85 const auto seq_length = input_vk.size(1); in lstm_input() local
341 const auto seq_length = input_vk.size(1); in run_lstm_context() local
|
| /aosp_15_r20/external/executorch/extension/llm/custom_ops/ |
| H A D | op_update_quantized_cache.cpp | 27 int64_t seq_length) { in validate_cache_params()
|
| H A D | op_sdpa.cpp | 649 int64_t seq_length) { in validate_cache_params()
701 int64_t seq_length) { // NOLINT: unused parameter 'seq_length' in update_cache()
|
| /aosp_15_r20/external/icu/icu4c/source/i18n/ |
| H A D | csr2022.cpp | 49 int32_t seq_length = (int32_t)uprv_strlen((const char *) seq); in match_2022() local
|
| /aosp_15_r20/external/cronet/third_party/icu/source/i18n/ |
| H A D | csr2022.cpp | 49 int32_t seq_length = (int32_t)uprv_strlen((const char *) seq); in match_2022() local
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_padding.py | 43 def gen_transformer_inputs(vocab_size, bs, seq_length): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | ctc_ops.py | 1124 def collapse_repeated(labels, seq_length, name=None): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | RNN.cpp | 87 int64_t seq_length; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.cc | 1918 MIOpenRnnSequenceTensorDescriptor(int seq_length, int batch_size, in MIOpenRnnSequenceTensorDescriptor()
1956 int seq_length() const { return seq_length_; } in seq_length() function in stream_executor::gpu::MIOpenRnnSequenceTensorDescriptor
2019 int seq_length = 0; member
2604 MIOpenSupport::createRnnSequenceTensorDescriptor(int seq_length, int batch_size, in createRnnSequenceTensorDescriptor()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | RNN_miopen.cpp | 160 int64_t seq_length; member
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | shape_type_inference.cpp | 1182 c10::ShapeSymbol seq_length; in ProcessTimeSeriesNode() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | RNN.cpp | 425 int64_t seq_length; member
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 7086 int64_t seq_length = input.size(0); in mkldnn_rnn_layer_differentiable_backward() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_jit.py | 257 def get_lstm_inputs(device, training=False, seq_length=None): argument
|