| 118 | void SetBias(std::initializer_list<float> f) { PopulateTensor(bias_, f); } |
| 119 | |
| 120 | void SetInput(std::initializer_list<float> data) { |
| 121 | PopulateTensor(input_, data); |
| 122 | } |
| 123 | std::vector<float> GetOutput() { return ExtractVector<float>(output_); } |
| 124 | }; |
| 125 |