Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add TopK op #5839

Open
wants to merge 31 commits into
base: master
Choose a base branch
from
+1,960 −1,094
Open

add TopK op #5839

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
31 commits
Select commit Hold shift + click to select a range
f1db785
c_api surpport set_vulkan_device
Baiyuetribe Jul 31, 2024
ae2f9f0
修复无vulkan编辑条件下的编译错误
Baiyuetribe Jul 31, 2024
0278590
fix action queen
Baiyuetribe Aug 1, 2024
fea7343
add NCNN_VULKAN option
Baiyuetribe Aug 1, 2024
0109929
remove space
Baiyuetribe Aug 1, 2024
90a4161
Merge branch 'Tencent:master' into master
Baiyuetribe Dec 21, 2024
0e6a38c
add TopK op
Baiyuetribe Dec 21, 2024
08c330a
fix type error
Baiyuetribe Dec 21, 2024
13dacbb
apply code-format changes
Baiyuetribe Dec 21, 2024
7f732af
重构 TopK 类中的排序逻辑，使用结构体替代 Lambda 表达式以提高兼容性
Baiyuetribe Dec 21, 2024
2cb7408
Merge branch 'master' into master
Baiyuetribe Dec 31, 2024
c6edde6
fix gcc-arm64
Baiyuetribe Dec 31, 2024
5fa5d6e
ref argmax，fix <vect>
Baiyuetribe Dec 31, 2024
b3c3a90
fix linux-cpp-with-simstl
Baiyuetribe Dec 31, 2024
b6d9fb1
linux-clang-simstl with no functional lib
Baiyuetribe Dec 31, 2024
f4835c5
add simplestl env
Baiyuetribe Dec 31, 2024
77f54cb
add simpstl env for resize
Baiyuetribe Dec 31, 2024
c1cf4e9
fix vs2015
Baiyuetribe Dec 31, 2024
96d0745
fix pnnx not found path
Baiyuetribe Dec 31, 2024
8422853
搞不定dims=4,axis=0和1的情况
Baiyuetribe Jan 4, 2025
f1af190
fix pnnx
Baiyuetribe Jan 4, 2025
2308adc
clean code
Baiyuetribe Jan 6, 2025
0395de9
Merge branch 'master' into master
Baiyuetribe Jan 6, 2025
87eb2d0
fix ctest
Baiyuetribe Jan 6, 2025
cb91ef7
remove 4d axis=0|1
Baiyuetribe Jan 6, 2025
f3342d5
fix pnnx path
Baiyuetribe Jan 6, 2025
2440a7f
fix c++03
Baiyuetribe Jan 6, 2025
82bb4ee
fixed index to i32
Baiyuetribe Jan 7, 2025
ffb1daf
Merge branch 'master' into master
Baiyuetribe Jan 7, 2025
3afcda7
Merge branch 'master' into master
Baiyuetribe Jan 7, 2025
5fa6f5b
fix 4D with d and c
Baiyuetribe Jan 12, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2,314 changes: 1,220 additions & 1,094 deletions docs/developer-guide/operators.md
View file
Open in desktop

Large diffs are not rendered by default.

1 change: 1 addition & 0 deletions src/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -169,6 +169,7 @@ ncnn_add_layer(Shrink)
ncnn_add_layer(RMSNorm)
ncnn_add_layer(Spectrogram)
ncnn_add_layer(InverseSpectrogram)
ncnn_add_layer(TopK)

if(NCNN_VULKAN)
ncnn_add_shader(${CMAKE_CURRENT_SOURCE_DIR}/convert_ycbcr.comp)
Expand Down
Loading
Loading