/*!
 * 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