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 '0c2790f7e5'
${ noResults }
Paddle/paddle/operators/softmax_op.cu

6 lines
191 B
Raw Normal View History Unescape

replace Tensor::tensor to EigenTensor::From
8 years ago
#include "paddle/framework/op_registry.h"
#include "paddle/operators/softmax_op.h"
Add skeletons of `mul`, `rowwise_add`, `sigmoid`, `softmax` ops * Implement InferShape and register them, give a stub Kernel method by LOG(INFO)
8 years ago
REGISTER_OP_GPU_KERNEL(
implement some basic OpKernel
8 years ago
softmax, paddle::operators::SoftmaxKernel<paddle::platform::GPUPlace, float>);