plugin/torch/torch_criterion.cu (16 lines of code) (raw):
/*!
* Copyright (c) 2015 by Contributors
* \file activation.cc
* \brief activation op
* \author Bing Xu
*/
#include "./torch_criterion-inl.h"
#include "../../src/operator/mshadow_op.h"
namespace mxnet {
namespace op {
template<>
Operator *CreateOp<gpu>(TorchCriterionParam param) {
return new TorchCriterionOp<gpu>(param);
}
} // namespace op
} // namespace mxnet