Searched refs:TopContainer (Results 1 – 1 of 1) sorted by relevance
83 class TopContainer { class85 TopContainer() = delete;86 TopContainer(int32 k, int32 row_size) : k_(k) { in TopContainer() function in tflite::ops::builtin::topk_v2::__anonb35a0bd70111::TopContainer140 TopContainer<T> topc(k, row_size); in TopK()