experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Lite: Range Op small refactor

Browse Source
This commit is contained in:
ANSHUMAN TRIPATHY 2019-03-25 18:35:26 +05:30
parent f7e277d66d
commit 70a2406447
1 changed files with 1 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tensorflow/lite/kernels/range.cc
View File

@ -93,18 +93,7 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
TF_LITE_ENSURE_EQ(context, delta->type, dtype); TF_LITE_ENSURE_EQ(context, delta->type, dtype);
TfLiteTensor* output = GetOutput(context, node, kOutputTensor); TfLiteTensor* output = GetOutput(context, node, kOutputTensor);
switch (dtype) { output->type = 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;
}
if (IsConstantTensor(start) && IsConstantTensor(limit) && if (IsConstantTensor(start) && IsConstantTensor(limit) &&
IsConstantTensor(delta)) { IsConstantTensor(delta)) {