Searched defs:sequence_len (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A 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::__anon6f5dbcbd0111::UnidirectionalRNNOpModel
251 HybridUnidirectionalRNNOpModel(int batches, int sequence_len, int units, in HybridUnidirectionalRNNOpModel()
|
| H A 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::__anona5c2041c0111::BidirectionalRNNOpModel
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | SDPA.cpp | 189 const ValueRef sequence_len = args[arg_idx++]; in sdpa_with_kv_cache_impl() local
|
| /aosp_15_r20/external/executorch/backends/vulkan/test/op_tests/ |
| H A D | sdpa_test.cpp | 215 const int sequence_len, in test_reference_sdpa()
|
| /aosp_15_r20/hardware/google/gfxstream/third-party/astc-encoder/Source/ |
| D | astcenc_find_best_partitioning.cpp | 565 …unsigned int sequence_len = compute_kmeans_partition_ordering(bsd, blk, partition_count, partition… in find_best_partition_candidates() local
|