Searched defs:sequence_len (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_rnn_test.cc | 174 int batches, int sequence_len, int units, int size, bool time_major, in UnidirectionalRNNOpModel()
232 int sequence_len() { return sequence_len_; } in sequence_len() function in tflite::__anon8708873d0111::UnidirectionalRNNOpModel
251 HybridUnidirectionalRNNOpModel(int batches, int sequence_len, int units, in HybridUnidirectionalRNNOpModel()
|
| D | bidirectional_sequence_rnn_test.cc | 665 BidirectionalRNNOpModel(int batches, int sequence_len, int fw_units, in BidirectionalRNNOpModel()
821 int sequence_len() { return sequence_len_; } in sequence_len() function in tflite::__anonb5fd7e9c0111::BidirectionalRNNOpModel
|
| /external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| D | SDPA.cpp | 189 const ValueRef sequence_len = args[arg_idx++]; in sdpa_with_kv_cache_impl() local
|
| /external/executorch/backends/vulkan/test/op_tests/ |
| D | sdpa_test.cpp | 215 const int sequence_len, in test_reference_sdpa()
|