| 312 | } |
| 313 | |
| 314 | void SetField(const std::string& name, const std::string& value) override { |
| 315 | if (name == "input") { |
| 316 | if (expected_inputs_.empty()) { |
| 317 | return test_runner_->Invalidate("Too many inputs to reshape"); |
| 318 | } |
| 319 | test_runner_->ReshapeTensor(*expected_inputs_.begin(), value); |
| 320 | expected_inputs_.erase(expected_inputs_.begin()); |
| 321 | } |
| 322 | } |
| 323 | |
| 324 | private: |
| 325 | std::vector<int> expected_inputs_; |
nothing calls this directly
no test coverage detected