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

Method configure_layer_norm

src/runtime/CL/functions/CLQLSTMLayer.cpp:122–131  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

120CLQLSTMLayer::~CLQLSTMLayer() = default;
121
122void CLQLSTMLayer::configure_layer_norm(LayerNormGate g, const ICLTensor *in)
123{
124 ARM_COMPUTE_ERROR_ON(!_has_layer_norm);
125
126 CLTensor *out = &get_layer_norm_output(g);
127 _memory_group.manage(out);
128 out->allocator()->init(*(in->info()));
129
130 get_layer_norm(g).configure(in, out, get_layer_norm_weight(g), get_layer_norm_bias(g));
131}
132
133Status CLQLSTMLayer::validate_layer_norm(const ITensorInfo &in, const ITensorInfo &weight, const ITensorInfo &bias)
134{

Callers

nothing calls this directly

Calls 5

manageMethod · 0.45
initMethod · 0.45
allocatorMethod · 0.45
infoMethod · 0.45
configureMethod · 0.45

Tested by

no test coverage detected