| 186 | }; |
| 187 | |
| 188 | class SmallLSTMLayerDataset final : public LSTMLayerDataset |
| 189 | { |
| 190 | public: |
| 191 | SmallLSTMLayerDataset() |
| 192 | { |
| 193 | add_config(TensorShape(8U), TensorShape(8U, 16U), TensorShape(16U, 16U), TensorShape(16U), TensorShape(16U), |
| 194 | TensorShape(16U), TensorShape(64U), |
| 195 | ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU), 0.05f, 0.93f); |
| 196 | add_config(TensorShape(8U, 2U), TensorShape(8U, 16U), TensorShape(16U, 16U), TensorShape(16U), |
| 197 | TensorShape(16U, 2U), TensorShape(16U, 2U), TensorShape(64U, 2U), |
| 198 | ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU), 0.05f, 0.93f); |
| 199 | add_config(TensorShape(8U, 2U), TensorShape(8U, 16U), TensorShape(16U, 16U), TensorShape(16U), |
| 200 | TensorShape(16U, 2U), TensorShape(16U, 2U), TensorShape(48U, 2U), |
| 201 | ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU), 0.05f, 0.93f); |
| 202 | } |
| 203 | }; |
| 204 | |
| 205 | } // namespace datasets |
| 206 | } // namespace test |
no outgoing calls
no test coverage detected