tfopgen icon indicating copy to clipboard operation
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...