diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-11 19:38:48 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-11 19:41:29 -0700 |
commit | 52e2698ac969a0f82c6ce901f80f04818ca8ac4e (patch) | |
tree | 5b89fa879c61cac2f3b64d5edbf405ad616edde5 /tensorflow/contrib/lite/kernels/svdf.cc | |
parent | 84b5938aaee991d6909e16e56c66bf88e8843fbb (diff) |
Making GetInput from kernel_util.h return a pointer to const data.
PiperOrigin-RevId: 196340200
Diffstat (limited to 'tensorflow/contrib/lite/kernels/svdf.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/svdf.cc | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/tensorflow/contrib/lite/kernels/svdf.cc b/tensorflow/contrib/lite/kernels/svdf.cc index 13da51c7a7..788812755e 100644 --- a/tensorflow/contrib/lite/kernels/svdf.cc +++ b/tensorflow/contrib/lite/kernels/svdf.cc @@ -58,9 +58,10 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { TF_LITE_ENSURE_EQ(context, node->outputs->size, 2); TfLiteTensor* input = &context->tensors[node->inputs->data[kInputTensor]]; - TfLiteTensor* weights_feature = + const TfLiteTensor* weights_feature = GetInput(context, node, kWeightsFeatureTensor); - TfLiteTensor* weights_time = GetInput(context, node, kWeightsTimeTensor); + const TfLiteTensor* weights_time = + GetInput(context, node, kWeightsTimeTensor); // Check all the parameters of tensor match within themselves and match the // input configuration. @@ -123,10 +124,11 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { auto* params = reinterpret_cast<TfLiteSVDFParams*>(node->builtin_data); - TfLiteTensor* input = GetInput(context, node, kInputTensor); - TfLiteTensor* weights_feature = + const TfLiteTensor* input = GetInput(context, node, kInputTensor); + const TfLiteTensor* weights_feature = GetInput(context, node, kWeightsFeatureTensor); - TfLiteTensor* weights_time = GetInput(context, node, kWeightsTimeTensor); + const TfLiteTensor* weights_time = + GetInput(context, node, kWeightsTimeTensor); TfLiteTensor* state = GetOutput(context, node, kStateTensor); TfLiteTensor* output = GetOutput(context, node, kOutputTensor); |