• Home
Name Date Size #Lines LOC

..--

BUILDD03-May-20243.3 KiB160142

README.mdD03-May-2024116 32

__init__.pyD03-May-20240 10

attr_examples.ccD03-May-20241.6 KiB4723

cuda_op.pyD03-May-2024962 247

cuda_op_kernel.ccD03-May-20241.9 KiB5627

cuda_op_kernel.cu.ccD03-May-20241.3 KiB3516

cuda_op_test.pyD03-May-20241 KiB3110

fact_test.pyD03-May-20241 KiB3110

zero_out_1_test.pyD03-May-20241.8 KiB5225

zero_out_2_test.pyD03-May-20241.8 KiB5023

zero_out_3_test.pyD03-May-20241.5 KiB4318

zero_out_grad_2.pyD03-May-20241.5 KiB4120

zero_out_op_1.pyD03-May-20241.1 KiB269

zero_out_op_2.pyD03-May-20241 KiB269

zero_out_op_3.pyD03-May-2024953 247

zero_out_op_kernel_1.ccD03-May-20243.1 KiB9762

zero_out_op_kernel_2.ccD03-May-20243.7 KiB11778

zero_out_op_kernel_3.ccD03-May-20242.6 KiB7340

README.md

1This is the code as described in https://tensorflow.org/guide/create_op.
2See `../custom_ops_doc` for more examples.
3