diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-16 20:31:29 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-16 20:34:29 -0700 |
commit | b2e53b91019f9ab00fe133fe10b2d29bc7e5886c (patch) | |
tree | 4c22811166bf58ac47668d18d0b5ff6685b53a56 /tensorflow/contrib/lite/kernels/kernel_util.h | |
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/kernel_util.h')
-rw-r--r-- | tensorflow/contrib/lite/kernels/kernel_util.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/kernels/kernel_util.h b/tensorflow/contrib/lite/kernels/kernel_util.h index de0e368891..82cded36f2 100644 --- a/tensorflow/contrib/lite/kernels/kernel_util.h +++ b/tensorflow/contrib/lite/kernels/kernel_util.h @@ -47,8 +47,9 @@ inline int64_t NumElements(const TfLiteTensor* t) { return count; } -inline TfLiteTensor* GetOptionalInputTensor(TfLiteContext* context, - const TfLiteNode* node, int index) { +inline const TfLiteTensor* GetOptionalInputTensor(TfLiteContext* context, + const TfLiteNode* node, + int index) { const bool use_tensor = node->inputs->data[index] != kOptionalTensor; if (use_tensor) { return &context->tensors[node->inputs->data[index]]; |