| 32 | |
| 33 | namespace { |
| 34 | TfLiteStatus ExpandTensorDim(TfLiteContext* context, const TfLiteTensor& input, |
| 35 | int axis, TfLiteTensor* output) { |
| 36 | const TfLiteIntArray& input_dims = *input.dims; |
| 37 | if (axis < 0) { |
| 38 | axis = input_dims.size + 1 + axis; |
| 39 | } |
| 40 | TF_LITE_ENSURE(context, axis <= input_dims.size); |
| 41 | |
| 42 | TfLiteIntArray* output_dims = TfLiteIntArrayCreate(input_dims.size + 1); |
| 43 | for (int i = 0; i < output_dims->size; ++i) { |
| 44 | if (i < axis) { |
| 45 | output_dims->data[i] = input_dims.data[i]; |
| 46 | } else if (i == axis) { |
| 47 | output_dims->data[i] = 1; |
| 48 | } else { |
| 49 | output_dims->data[i] = input_dims.data[i - 1]; |
| 50 | } |
| 51 | } |
| 52 | |
| 53 | return context->ResizeTensor(context, output, output_dims); |
| 54 | } |
| 55 | |
| 56 | TfLiteStatus GetAxisValueFromTensor(TfLiteContext* context, |
| 57 | const TfLiteTensor& axis, int* axis_value) { |
no test coverage detected