aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/kernels/kernel_util.h
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-05-03 23:36:02 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-05-04 10:43:07 -0700
commitda0dcb21501b765932e392ae710ebbecefeb309c (patch)
treeb0547999a958c98316cad7f6b90a14059569b733 /tensorflow/contrib/lite/kernels/kernel_util.h
parent8ec11ae8eb7b97caced73ed3971209236e2aef5c (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.h4
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; }