| 54 | } |
| 55 | |
| 56 | std::vector<int> FlexModelTest::GetShape(int tensor_index) { |
| 57 | std::vector<int> result; |
| 58 | auto* dims = interpreter_->tensor(tensor_index)->dims; |
| 59 | result.reserve(dims->size); |
| 60 | for (int i = 0; i < dims->size; ++i) { |
| 61 | result.push_back(dims->data[i]); |
| 62 | } |
| 63 | return result; |
| 64 | } |
| 65 | |
| 66 | TfLiteType FlexModelTest::GetType(int tensor_index) { |
| 67 | return interpreter_->tensor(tensor_index)->type; |