add_relu.cpp 360 Bytes
Newer Older
Paul's avatar
Paul committed
1
2
3
4
5
6
7
8
9
#include <migraph/gpu/device/contiguous.hpp>
#include <migraph/gpu/device/binary.hpp>

namespace migraph {
namespace gpu {
namespace device {

void add_relu(argument arg1, argument arg2, argument result)
{
Paul's avatar
Paul committed
10
    binary_standard(arg1, arg2, result, [](auto x, auto y) { return max(0, x + y); });
Paul's avatar
Paul committed
11
12
13
14
15
}

} // namespace device
} // namespace gpu
} // namespace migraph