diff options
author | 2018-09-26 13:42:36 -0700 | |
---|---|---|
committer | 2018-09-26 13:47:11 -0700 | |
commit | 652ce1aaefdadd04a9905a0788ab26c6fff93658 (patch) | |
tree | 58a766a41069b3985de75ab092d3672d4d737a4d /tensorflow/contrib/lite/kernels/pack.cc | |
parent | d600b1b55fa851648918fed7a67f61eefd554034 (diff) |
Kernel signature reworking, misc kernel improvements and migrations.
PiperOrigin-RevId: 214661332
Diffstat (limited to 'tensorflow/contrib/lite/kernels/pack.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/pack.cc | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/tensorflow/contrib/lite/kernels/pack.cc b/tensorflow/contrib/lite/kernels/pack.cc index 4cb98fdd19..c368582ef7 100644 --- a/tensorflow/contrib/lite/kernels/pack.cc +++ b/tensorflow/contrib/lite/kernels/pack.cc @@ -85,9 +85,12 @@ template <typename T> void PackImpl(TfLiteContext* context, TfLiteNode* node, TfLiteTensor* output, int values_count, int axis) { VectorOfTensors<T> all_inputs(*context, *node->inputs); - reference_ops::Pack<T>(RemapDim(NumDimensions(output), axis), - all_inputs.data(), all_inputs.dims(), values_count, - GetTensorData<T>(output), GetTensorDims(output)); + tflite::PackParams op_params; + op_params.axis = axis; + op_params.inputs_count = values_count; + + reference_ops::Pack<T>(op_params, all_inputs.shapes(), all_inputs.data(), + GetTensorShape(output), GetTensorData<T>(output)); } TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { |