/*! | |
* Copyright (c) 2015 by Contributors | |
* \file loss_binary_op.cu | |
* \brief loss function that takes a data and label | |
*/ | |
#include "./loss_binary_op-inl.h" | |
namespace mxnet { | |
namespace op { | |
NNVM_REGISTER_OP(softmax_cross_entropy) | |
.set_attr<FCompute>("FCompute<gpu>", SoftmaxCrossEntropyForward<gpu>); | |
NNVM_REGISTER_OP(_backward_softmax_cross_entropy) | |
.set_attr<FCompute>("FCompute<gpu>", SoftmaxCrossEntropyBackward<gpu>); | |
} // namespace op | |
} // namespace mxnet |