| 80 | |
| 81 | template <typename T> |
| 82 | void UnpackImpl(TfLiteContext* context, TfLiteNode* node, |
| 83 | const TfLiteTensor* input, int output_count, int axis) { |
| 84 | tflite::UnpackParams op_params; |
| 85 | op_params.axis = axis; |
| 86 | op_params.num_split = output_count; |
| 87 | VectorOfTensors<T> all_outputs(*context, *node->outputs); |
| 88 | reference_ops::Unpack<T>(op_params, GetTensorShape(input), |
| 89 | GetTensorData<T>(input), **all_outputs.shapes(), |
| 90 | all_outputs.data()); |
| 91 | } |
| 92 | |
| 93 | TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { |
| 94 | const TfLiteUnpackParams* data = |
nothing calls this directly
no test coverage detected