├── README.md ├── kl_loss_layer.cpp ├── kl_loss_layer.cu └── kl_loss_layer.hpp /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuancoder222/KL-loss/HEAD/README.md -------------------------------------------------------------------------------- /kl_loss_layer.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuancoder222/KL-loss/HEAD/kl_loss_layer.cpp -------------------------------------------------------------------------------- /kl_loss_layer.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuancoder222/KL-loss/HEAD/kl_loss_layer.cu -------------------------------------------------------------------------------- /kl_loss_layer.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuancoder222/KL-loss/HEAD/kl_loss_layer.hpp --------------------------------------------------------------------------------