mean_op.cu 149 字节
Newer Older
L
liaogang 已提交
1 2 3 4
#include "paddle/framework/op_registry.h"
#include "paddle/operators/mean_op.h"

REGISTER_OP_GPU_KERNEL(mean, ops::AddKernel<ops::GPUPlace, float>);