| 136 | }; |
| 137 | |
| 138 | class SmallRNNLayerDataset final : public RNNLayerDataset |
| 139 | { |
| 140 | public: |
| 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 |
no outgoing calls
no test coverage detected