| 88 | } |
| 89 | |
| 90 | TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { |
| 91 | // Just copy input to output. |
| 92 | const TfLiteTensor* input = GetInput(context, node, kInput); |
| 93 | TfLiteTensor* output = GetOutput(context, node, 0); |
| 94 | const TfLiteTensor* axis = GetInput(context, node, kAxis); |
| 95 | if (IsDynamicTensor(output)) { |
| 96 | int axis_value; |
| 97 | TF_LITE_ENSURE_OK(context, |
| 98 | GetAxisValueFromTensor(context, *axis, &axis_value)); |
| 99 | TF_LITE_ENSURE_OK(context, |
| 100 | ExpandTensorDim(context, *input, axis_value, output)); |
| 101 | } |
| 102 | memcpy(output->data.raw, input->data.raw, input->bytes); |
| 103 | return kTfLiteOk; |
| 104 | } |
| 105 | |
| 106 | } // namespace expand_dims |
| 107 | TfLiteRegistration* Register_EXPAND_DIMS() { |
nothing calls this directly
no test coverage detected