diff options
author | 2018-05-16 20:31:29 -0700 | |
---|---|---|
committer | 2018-05-16 20:34:29 -0700 | |
commit | b2e53b91019f9ab00fe133fe10b2d29bc7e5886c (patch) | |
tree | 4c22811166bf58ac47668d18d0b5ff6685b53a56 /tensorflow/contrib/lite/kernels/fully_connected.cc | |
parent | 0668573009abcf80b84ae7096211e83ae6ca6477 (diff) |
Making GetOptionalInput from kernel_util.h return a pointer to const data.
PiperOrigin-RevId: 196932028
Diffstat (limited to 'tensorflow/contrib/lite/kernels/fully_connected.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/fully_connected.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/kernels/fully_connected.cc b/tensorflow/contrib/lite/kernels/fully_connected.cc index 39b108629a..1ba30649ec 100644 --- a/tensorflow/contrib/lite/kernels/fully_connected.cc +++ b/tensorflow/contrib/lite/kernels/fully_connected.cc @@ -91,7 +91,7 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { const TfLiteTensor* input = GetInput(context, node, kInputTensor); const TfLiteTensor* filter = GetInput(context, node, kWeightsTensor); - TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); + const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); TfLiteTensor* output = GetOutput(context, node, kOutputTensor); // Check all the parameters of tensor match within themselves and match the @@ -347,7 +347,7 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { const TfLiteTensor* input = GetInput(context, node, kInputTensor); const TfLiteTensor* filter = GetInput(context, node, kWeightsTensor); - TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); + const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); TfLiteTensor* output = GetOutput(context, node, kOutputTensor); switch (filter->type) { // Already know in/out types are same. |