diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-03 23:36:02 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-04 10:43:07 -0700 |
commit | da0dcb21501b765932e392ae710ebbecefeb309c (patch) | |
tree | b0547999a958c98316cad7f6b90a14059569b733 /tensorflow/contrib/lite/kernels/kernel_util.h | |
parent | 8ec11ae8eb7b97caced73ed3971209236e2aef5c (diff) |
Internal change.
PiperOrigin-RevId: 195374319
Diffstat (limited to 'tensorflow/contrib/lite/kernels/kernel_util.h')
-rw-r--r-- | tensorflow/contrib/lite/kernels/kernel_util.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/kernel_util.h b/tensorflow/contrib/lite/kernels/kernel_util.h index 2f407b5da3..e225443a67 100644 --- a/tensorflow/contrib/lite/kernels/kernel_util.h +++ b/tensorflow/contrib/lite/kernels/kernel_util.h @@ -32,6 +32,10 @@ inline TfLiteTensor* GetOutput(TfLiteContext* context, TfLiteNode* node, int index) { return &context->tensors[node->outputs->data[index]]; } +inline TfLiteTensor* GetTemporary(TfLiteContext* context, TfLiteNode* node, + int index) { + return &context->tensors[node->temporaries->data[index]]; +} inline int NumInputs(const TfLiteNode* node) { return node->inputs->size; } inline int NumOutputs(const TfLiteNode* node) { return node->outputs->size; } |