├── .gitignore ├── CMakeLists.txt ├── LICENSE ├── README.md ├── check_op.py ├── common.cc ├── common.h ├── l_softmax_grad_op.cc ├── l_softmax_grad_op.cu ├── l_softmax_op.cc ├── l_softmax_op.cu ├── l_softmax_op.h ├── py_loss.py ├── test_op.py └── work_sharder.h /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HiKapok/tf.extra_losses/HEAD/.gitignore -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HiKapok/tf.extra_losses/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HiKapok/tf.extra_losses/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HiKapok/tf.extra_losses/HEAD/README.md -------------------------------------------------------------------------------- /check_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HiKapok/tf.extra_losses/HEAD/check_op.py -------------------------------------------------------------------------------- /common.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HiKapok/tf.extra_losses/HEAD/common.cc -------------------------------------------------------------------------------- /common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HiKapok/tf.extra_losses/HEAD/common.h -------------------------------------------------------------------------------- /l_softmax_grad_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HiKapok/tf.extra_losses/HEAD/l_softmax_grad_op.cc -------------------------------------------------------------------------------- /l_softmax_grad_op.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HiKapok/tf.extra_losses/HEAD/l_softmax_grad_op.cu -------------------------------------------------------------------------------- /l_softmax_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HiKapok/tf.extra_losses/HEAD/l_softmax_op.cc -------------------------------------------------------------------------------- /l_softmax_op.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HiKapok/tf.extra_losses/HEAD/l_softmax_op.cu -------------------------------------------------------------------------------- /l_softmax_op.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HiKapok/tf.extra_losses/HEAD/l_softmax_op.h -------------------------------------------------------------------------------- /py_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HiKapok/tf.extra_losses/HEAD/py_loss.py -------------------------------------------------------------------------------- /test_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HiKapok/tf.extra_losses/HEAD/test_op.py -------------------------------------------------------------------------------- /work_sharder.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HiKapok/tf.extra_losses/HEAD/work_sharder.h --------------------------------------------------------------------------------