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

Class SmallRNNLayerDataset

tests/datasets/RNNLayerDataset.h:138–146  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

136};
137
138class SmallRNNLayerDataset final : public RNNLayerDataset
139{
140public:
141 SmallRNNLayerDataset()
142 {
143 add_config(TensorShape(128U, 16U), TensorShape(128U, 32U), TensorShape(32U, 32U), TensorShape(32U),
144 TensorShape(32U, 16U), ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU));
145 }
146};
147
148} // namespace datasets
149} // namespace test

Callers 2

RNNLayer.cppFile · 0.85
RNNLayer.cppFile · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected