| 85 | |
| 86 | template <typename T> |
| 87 | TfLiteStatus PackImpl(TfLiteContext* context, TfLiteNode* node, |
| 88 | TfLiteTensor* output, int values_count, int axis) { |
| 89 | TF_LITE_ENSURE(context, axis >= 0); |
| 90 | |
| 91 | VectorOfTensors<T> all_inputs(*context, *node->inputs); |
| 92 | tflite::PackParams op_params; |
| 93 | op_params.axis = axis; |
| 94 | op_params.inputs_count = values_count; |
| 95 | |
| 96 | reference_ops::Pack<T>(op_params, all_inputs.shapes(), all_inputs.data(), |
| 97 | GetTensorShape(output), GetTensorData<T>(output)); |
| 98 | return kTfLiteOk; |
| 99 | } |
| 100 | |
| 101 | TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { |
| 102 | const TfLitePackParams* data = |
nothing calls this directly
no test coverage detected