add_relu.cpp 373 Bytes
Newer Older
Paul's avatar
Paul committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
#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)
{
    binary_standard(arg1, arg2, result, [](auto x, auto y) { 
        return max(0, x + y);
    });
}

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