diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-08-30 11:17:57 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-30 11:23:53 -0700 |
commit | 9e12f1df3270b5e0b310645e6c3cae9fbd3f5dfc (patch) | |
tree | 6fb67b08ce4747aaf27f40d71a42edab04ea176c /tensorflow/contrib/lite/kernels/space_to_depth.cc | |
parent | 35bae087dce1e88c66007907f9e1b6b5b2958f10 (diff) |
Consolidate refactoring of runtime shapes.
PiperOrigin-RevId: 210945714
Diffstat (limited to 'tensorflow/contrib/lite/kernels/space_to_depth.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/space_to_depth.cc | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/tensorflow/contrib/lite/kernels/space_to_depth.cc b/tensorflow/contrib/lite/kernels/space_to_depth.cc index 9dbe9b9eda..9238e879f8 100644 --- a/tensorflow/contrib/lite/kernels/space_to_depth.cc +++ b/tensorflow/contrib/lite/kernels/space_to_depth.cc @@ -79,10 +79,12 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { const TfLiteTensor* input = GetInput(context, node, kInputTensor); TfLiteTensor* output = GetOutput(context, node, kOutputTensor); -#define TF_LITE_SPACE_TO_DEPTH(type, scalar) \ - type::SpaceToDepth<scalar>( \ - GetTensorData<scalar>(input), GetTensorDims(input), params->block_size, \ - GetTensorData<scalar>(output), GetTensorDims(output)) +#define TF_LITE_SPACE_TO_DEPTH(type, scalar) \ + tflite::SpaceToDepthParams op_params; \ + op_params.block_size = params->block_size; \ + type::SpaceToDepth(op_params, GetTensorShape(input), \ + GetTensorData<scalar>(input), GetTensorShape(output), \ + GetTensorData<scalar>(output)) switch (input->type) { // Already know in/out types are same. case kTfLiteFloat32: if (kernel_type == kReference) { |