Make cast op support bool (#6562)

Also add `elemwise_sub/mul/abs/clip` layers
del_some_in_makelist
Yu Yang 7 years ago committed by GitHub
parent d069f2ca0a
commit 0a8addf802
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -74,4 +74,5 @@ REGISTER_OP_WITH_KERNEL(cast, ops::CastOpGradMaker, ops::CastOpInferShape,
REGISTER_OP_CPU_KERNEL(cast, ops::CastOpKernel<CPU, float>,
ops::CastOpKernel<CPU, double>,
ops::CastOpKernel<CPU, int>,
ops::CastOpKernel<CPU, int64_t>);
ops::CastOpKernel<CPU, int64_t>,
ops::CastOpKernel<CPU, bool>);

@ -19,4 +19,5 @@ using CastOpKernel =
paddle::operators::CastOpKernel<paddle::platform::CUDADeviceContext, T>;
REGISTER_OP_CUDA_KERNEL(cast, CastOpKernel<float>, CastOpKernel<double>,
CastOpKernel<int>, CastOpKernel<int64_t>);
CastOpKernel<int>, CastOpKernel<int64_t>,
CastOpKernel<bool>);

@ -15,8 +15,9 @@ __all__ = [
]
_REGISTER_LAYER_FROM_OPS = [
'mean', 'mul', 'elementwise_add', 'elementwise_div', 'dropout', 'reshape',
'sigmoid', 'scale', 'transpose', 'sigmoid_cross_entropy_with_logits'
'mean', 'mul', 'dropout', 'reshape', 'sigmoid', 'scale', 'transpose',
'sigmoid_cross_entropy_with_logits', 'elementwise_add', 'elementwise_div',
'elementwise_sub', 'elementwise_mul', 'clip', 'abs'
]
for _OP in set(_REGISTER_LAYER_FROM_OPS):

Loading…
Cancel
Save