m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from 'dd249a50a5'
${ noResults }
Paddle/paddle/operators/mean_op.cu

6 lines
208 B
Raw Normal View History Unescape

Add mean op unit test in python
8 years ago
#define EIGEN_USE_GPU
Add mean cpu/gpu op
8 years ago
#include "paddle/operators/mean_op.h"
Add mean op unit test in python
8 years ago
REGISTER_OP_GPU_KERNEL(mean, ops::MeanKernel<ops::GPUPlace, float>);
Add Gradient Operator for mean
8 years ago
REGISTER_OP_GPU_KERNEL(mean_grad, ops::MeanGradKernel<ops::GPUPlace, float>);