diff options
Diffstat (limited to 'arm_compute/graph/LayerDescriptors.h')
-rw-r--r-- | arm_compute/graph/LayerDescriptors.h | 54 |
1 files changed, 45 insertions, 9 deletions
diff --git a/arm_compute/graph/LayerDescriptors.h b/arm_compute/graph/LayerDescriptors.h index d8e6a6a87b..d632ed9e78 100644 --- a/arm_compute/graph/LayerDescriptors.h +++ b/arm_compute/graph/LayerDescriptors.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2019-2020 ARM Limited. + * Copyright (c) 2019-2020 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -37,8 +37,7 @@ namespace descriptors struct ConcatLayerDescriptor { /** Default constructor */ - ConcatLayerDescriptor() - : axis(DataLayoutDimension::CHANNEL), output_qinfo() + ConcatLayerDescriptor() : axis(DataLayoutDimension::CHANNEL), output_qinfo() { } @@ -46,8 +45,7 @@ struct ConcatLayerDescriptor * * @param[in] axis Axis. */ - ConcatLayerDescriptor(DataLayoutDimension axis) - : axis(axis), output_qinfo() + ConcatLayerDescriptor(DataLayoutDimension axis) : axis(axis), output_qinfo() { } @@ -76,9 +74,16 @@ struct EltwiseLayerDescriptor * @param[in] r_policy (Optional) Rounding policy used for the operation. Defaults to @ref RoundingPolicy::TO_ZERO * @param[in] fused_activation (Optional) Fused activation information. Defaults to empty (identity) @ref ActivationLayerInfo */ - EltwiseLayerDescriptor(EltwiseOperation op, QuantizationInfo out_quant_info = QuantizationInfo(), ConvertPolicy c_policy = ConvertPolicy::SATURATE, RoundingPolicy r_policy = RoundingPolicy::TO_ZERO, + EltwiseLayerDescriptor(EltwiseOperation op, + QuantizationInfo out_quant_info = QuantizationInfo(), + ConvertPolicy c_policy = ConvertPolicy::SATURATE, + RoundingPolicy r_policy = RoundingPolicy::TO_ZERO, ActivationLayerInfo fused_activation = ActivationLayerInfo()) - : op(op), out_quant_info(out_quant_info), c_policy(c_policy), r_policy(r_policy), fused_activation(fused_activation) + : op(op), + out_quant_info(out_quant_info), + c_policy(c_policy), + r_policy(r_policy), + fused_activation(fused_activation) { } @@ -89,6 +94,37 @@ struct EltwiseLayerDescriptor ActivationLayerInfo fused_activation; /**< Fused activation info */ }; +/** Unary Elementwise layer descriptor */ +struct UnaryEltwiseLayerDescriptor +{ + /** Constructor + * + * @param[in] op Unary element-wise operation to perform + * @param[in] out_quant_info (Optional) Output quantization information. Defaults to empty @ref QuantizationInfo + * @param[in] c_policy (Optional) Convert policy used for the operation. Defaults to @ref ConvertPolicy::SATURATE + * @param[in] r_policy (Optional) Rounding policy used for the operation. Defaults to @ref RoundingPolicy::TO_ZERO + * @param[in] fused_activation (Optional) Fused activation information. Defaults to empty (identity) @ref ActivationLayerInfo + */ + UnaryEltwiseLayerDescriptor(UnaryEltwiseOperation op, + QuantizationInfo out_quant_info = QuantizationInfo(), + ConvertPolicy c_policy = ConvertPolicy::SATURATE, + RoundingPolicy r_policy = RoundingPolicy::TO_ZERO, + ActivationLayerInfo fused_activation = ActivationLayerInfo()) + : op(op), + out_quant_info(out_quant_info), + c_policy(c_policy), + r_policy(r_policy), + fused_activation(fused_activation) + { + } + + UnaryEltwiseOperation op; /**< Unary element-wise operation to perform */ + QuantizationInfo out_quant_info; /**< Output quantization information */ + ConvertPolicy c_policy; /**< Convert policy */ + RoundingPolicy r_policy; /**< Rounding policy */ + ActivationLayerInfo fused_activation; /**< Fused activation info */ +}; + /** Deconvolution layer descriptor */ struct DeconvolutionLayerDescriptor { @@ -105,7 +141,7 @@ struct DeconvolutionLayerDescriptor PadStrideInfo info; /**< Padding and stride information */ QuantizationInfo out_quant_info; /**< Output quantization information */ }; -} // namespace descriptor +} // namespace descriptors } // namespace graph } // namespace arm_compute -#endif /* ARM_COMPUTE_LAYER_DESCRIPTORS_H */
\ No newline at end of file +#endif /* ARM_COMPUTE_LAYER_DESCRIPTORS_H */ |