diff options
Diffstat (limited to 'tensorflow/contrib/lite/delegates/eager/kernel.cc')
-rw-r--r-- | tensorflow/contrib/lite/delegates/eager/kernel.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/delegates/eager/kernel.cc b/tensorflow/contrib/lite/delegates/eager/kernel.cc index f8467c7cb2..0ee4db1ffb 100644 --- a/tensorflow/contrib/lite/delegates/eager/kernel.cc +++ b/tensorflow/contrib/lite/delegates/eager/kernel.cc @@ -278,7 +278,7 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { TfLiteTensor* tensor = &context->tensors[tensor_index]; TF_LITE_ENSURE_OK( context, - CopyShape(context, buffer_map->GetTensor(tensor_index), tensor)); + CopyShapeAndType(context, buffer_map->GetTensor(tensor_index), tensor)); tensor->buffer_handle = tensor_index; tensor->data_is_stale = true; } |