|
|
|
@@ -71,6 +71,18 @@ ATTR_MAP(Relu6Grad) = EMPTY_ATTR_MAP; |
|
|
|
OUTPUT_MAP(Relu6Grad) = {{0, OUTPUT_DESC(backprops)}}; |
|
|
|
REG_ADPT_DESC(Relu6Grad, kNameReLU6Grad, ADPT_DESC(Relu6Grad)) |
|
|
|
|
|
|
|
// Softplus |
|
|
|
INPUT_MAP(Softplus) = {{1, INPUT_DESC(x)}}; |
|
|
|
ATTR_MAP(Softplus) = EMPTY_ATTR_MAP; |
|
|
|
OUTPUT_MAP(Softplus) = {{0, OUTPUT_DESC(y)}}; |
|
|
|
REG_ADPT_DESC(Softplus, kNameSoftplus, ADPT_DESC(Softplus)) |
|
|
|
|
|
|
|
// SoftplusGrad |
|
|
|
INPUT_MAP(SoftplusGrad) = {{1, INPUT_DESC(gradients)}, {2, INPUT_DESC(features)}}; |
|
|
|
ATTR_MAP(SoftplusGrad) = EMPTY_ATTR_MAP; |
|
|
|
OUTPUT_MAP(SoftplusGrad) = {{0, OUTPUT_DESC(backprops)}}; |
|
|
|
REG_ADPT_DESC(SoftplusGrad, kNameSoftplusGrad, ADPT_DESC(SoftplusGrad)) |
|
|
|
|
|
|
|
// ReluGrad |
|
|
|
INPUT_MAP(ReluGrad) = {{1, INPUT_DESC(gradients)}, {2, INPUT_DESC(features)}}; |
|
|
|
ATTR_MAP(ReluGrad) = EMPTY_ATTR_MAP; |
|
|
|
|