tfopgen
tfopgen copied to clipboard
Generate C++ and CUDA boilerplate for tensorflow custom operators
Results
1
tfopgen issues
Sort by
recently updated
recently updated
newest added
After compiling the "simple" example, custom.so is properly created: Makefile custom_op.h custom_op_cpu.h custom_op_gpu.cu custom_op_gpu.o custom.so custom_op_cpu.cpp custom_op_cpu.o custom_op_gpu.cuh test_custom.py But running test_custom.py resulted in a "./custom.so: undefined symbol" error as...