| 60 | } |
| 61 | |
| 62 | bool SingleOpModel::PopulateTensor(int index, std::vector<float>&& data) { |
| 63 | if (index >= inputs_.size() || |
| 64 | inputs_[index].shape.DimensionsProduct() != data.size()) { |
| 65 | return false; |
| 66 | } |
| 67 | inputs_[index].data = std::move(data); |
| 68 | return true; |
| 69 | } |
| 70 | |
| 71 | Status SingleOpModel::Invoke(const CompilationOptions& compile_options, |
| 72 | const RuntimeOptions& runtime_options, |