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

Removed check that resize is up only.

Browse Source 
Resize uses the same formula for upsampling and downsampling.

PiperOrigin-RevId: 343536972
Change-Id: I1f9ab05d3b6feba1646d81a0ba29b0fc832ad0e2
This commit is contained in:
Raman Sarokin 2020-11-20 11:59:39 -08:00 committed by TensorFlower Gardener
parent 6271251b7f
commit 18201420c5
1 changed files with 0 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
tensorflow/lite/delegates/gpu/common/model_builder.cc
View File

@ -1656,7 +1656,6 @@ class Resize2DOperationParser : public TFLiteOperationParser {
RETURN_IF_ERROR(CheckInputsOutputs(context, tflite_node,
/*runtime_inputs=*/1, /*outputs=*/1));
RETURN_IF_ERROR(CheckOnlyUpsamplingIsSupported(context, tflite_node));
bool align_corners;
RETURN_IF_ERROR(GetAlignCornersValue(tflite_node, &align_corners));
bool half_pixel_centers;
@ -1728,27 +1727,6 @@ class Resize2DOperationParser : public TFLiteOperationParser {
return absl::OkStatus();
}
absl::Status CheckOnlyUpsamplingIsSupported(const TfLiteContext* context,
const TfLiteNode* tflite_node) {
const auto* input = context->tensors + tflite_node->inputs->data[0];
const auto* output = context->tensors + tflite_node->outputs->data[0];
if (!input->dims || input->dims->size != 4) {
return absl::InvalidArgumentError("input.dims.size != 4");
}
if (!output->dims || output->dims->size != 4) {
return absl::InvalidArgumentError("output.dims.size != 4");
}
if (output->dims->data[1] < input->dims->data[1] ||
output->dims->data[2] < input->dims->data[2]) {
return absl::InvalidArgumentError(absl::StrCat(
"Only upsampling is supported, received output h,w = ",
output->dims->data[1], ",", output->dims->data[2],
" input h,w = ", input->dims->data[1], ",", input->dims->data[2]));
}
return absl::OkStatus();
}
SamplingType sampling_type_ = SamplingType::UNKNOWN;
};