提交 4f4514fe 编写于 作者: L liuqi

Change the activation opencl kernel parameter type to float.

上级 fff4d5fd
#include <common.h>
__kernel void activation(__read_only image2d_t input,
__private const DATA_TYPE relux_max_limit,
__private const DATA_TYPE prelu_alpha,
__private const float relux_max_limit,
__private const float prelu_alpha,
__write_only image2d_t output) {
const int ch_blk = get_global_id(0);
const int w = get_global_id(1);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册