Skip to content

Commit 9cd7f17

Browse files
authored
Add CPU and GPU of median_filter operator
1 parent f35fb69 commit 9cd7f17

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

CMakeLists.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -170,6 +170,8 @@ set(SOURCES
170170
src/ops/mean.cc
171171
src/ops/mean_cpu.cc
172172
src/ops/median_filter.cc
173+
src/ops/median_filter_cpu.cc
174+
src/ops/median_filter_gpu.cu
173175
src/ops/min_max.cc
174176
src/ops/mul.cc
175177
src/ops/multinomial.cc

0 commit comments

Comments
 (0)