From 70a2406447d3c3729a71ef36697b6da0f3cf5e34 Mon Sep 17 00:00:00 2001 From: ANSHUMAN TRIPATHY Date: Mon, 25 Mar 2019 18:35:26 +0530 Subject: [PATCH] Lite: Range Op small refactor --- tensorflow/lite/kernels/range.cc | 13 +------------ 1 file changed, 1 insertion(+), 12 deletions(-) diff --git a/tensorflow/lite/kernels/range.cc b/tensorflow/lite/kernels/range.cc index eefe5db1ece..d6fc10fd5a5 100644 --- a/tensorflow/lite/kernels/range.cc +++ b/tensorflow/lite/kernels/range.cc @@ -93,18 +93,7 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { TF_LITE_ENSURE_EQ(context, delta->type, dtype); TfLiteTensor* output = GetOutput(context, node, kOutputTensor); - switch (dtype) { - case kTfLiteInt32: - output->type = kTfLiteInt32; - break; - case kTfLiteFloat32: - output->type = kTfLiteFloat32; - break; - default: - context->ReportError(context, "Unknown index output data type: %d", - dtype); - return kTfLiteError; - } + output->type = dtype; if (IsConstantTensor(start) && IsConstantTensor(limit) && IsConstantTensor(delta)) {