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/batch_to_space_nd.cc | |
parent | 35bae087dce1e88c66007907f9e1b6b5b2958f10 (diff) |
Consolidate refactoring of runtime shapes.
PiperOrigin-RevId: 210945714
Diffstat (limited to 'tensorflow/contrib/lite/kernels/batch_to_space_nd.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/batch_to_space_nd.cc | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/tensorflow/contrib/lite/kernels/batch_to_space_nd.cc b/tensorflow/contrib/lite/kernels/batch_to_space_nd.cc index c8cee88edf..4efa9d596d 100644 --- a/tensorflow/contrib/lite/kernels/batch_to_space_nd.cc +++ b/tensorflow/contrib/lite/kernels/batch_to_space_nd.cc @@ -125,14 +125,14 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { } #define TF_LITE_BATCH_TO_SPACE_ND(type, scalar) \ - type::BatchToSpaceND(GetTensorData<scalar>(op_context.input), \ - GetTensorDims(op_context.input), \ + type::BatchToSpaceND(GetTensorShape(op_context.input), \ + GetTensorData<scalar>(op_context.input), \ + GetTensorShape(op_context.block_shape), \ GetTensorData<int32_t>(op_context.block_shape), \ - GetTensorDims(op_context.block_shape), \ + GetTensorShape(op_context.crops), \ GetTensorData<int32_t>(op_context.crops), \ - GetTensorDims(op_context.crops), \ - GetTensorData<scalar>(op_context.output), \ - GetTensorDims(op_context.output)) + GetTensorShape(op_context.output), \ + GetTensorData<scalar>(op_context.output)) switch (op_context.input->type) { // Already know in/out types are same. case kTfLiteFloat32: if (kernel_type == kReference) { |