| 27 | constexpr int kOutputTensor = 0; |
| 28 | |
| 29 | TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { |
| 30 | const TfLiteTensor* input = GetInput(context, node, kInputTensor); |
| 31 | TfLiteTensor* output = GetOutput(context, node, kOutputTensor); |
| 32 | TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); |
| 33 | TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); |
| 34 | TF_LITE_ENSURE_EQ(context, input->type, kTfLiteFloat32); |
| 35 | output->type = input->type; |
| 36 | TfLiteIntArray* output_size = TfLiteIntArrayCopy(input->dims); |
| 37 | return context->ResizeTensor(context, output, output_size); |
| 38 | } |
| 39 | |
| 40 | TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { |
| 41 | const TfLiteTensor* input = GetInput(context, node, kInputTensor); |
nothing calls this directly
no test coverage detected