MCPcopy Create free account
hub / github.com/ARM-software/ComputeLibrary / NormalizationLayer

Class NormalizationLayer

arm_compute/graph/frontend/Layers.h:881–901  ·  view source on GitHub ↗

Normalization Layer */

Source from the content-addressed store, hash-verified

879
880/** Normalization Layer */
881class NormalizationLayer final : public ILayer
882{
883public:
884 /** Construct a normalization layer.
885 *
886 * @param[in] norm_info Normalization information.
887 */
888 NormalizationLayer(NormalizationLayerInfo norm_info) : _norm_info(norm_info)
889 {
890 }
891
892 NodeID create_layer(IStream &s) override
893 {
894 NodeParams common_params = {name(), s.hints().target_hint};
895 NodeIdxPair input = {s.tail_node(), 0};
896 return GraphBuilder::add_normalization_node(s.graph(), common_params, input, _norm_info);
897 }
898
899private:
900 NormalizationLayerInfo _norm_info;
901};
902
903/** Normalize planar YUV Layer */
904class NormalizePlanarYUVLayer final : public ILayer

Callers 2

do_setupMethod · 0.85
do_setupMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected