diff --git a/tensorflow/lite/delegates/gpu/common/model_builder.cc b/tensorflow/lite/delegates/gpu/common/model_builder.cc index 7950f10c642..95e330a5544 100644 --- a/tensorflow/lite/delegates/gpu/common/model_builder.cc +++ b/tensorflow/lite/delegates/gpu/common/model_builder.cc @@ -2265,6 +2265,7 @@ class TransformLandmarksOperationParser : public TFLiteOperationParser { absl::Status IsSupported(const TfLiteContext* context, const TfLiteNode* tflite_node, const TfLiteRegistration* registration) final { + RETURN_IF_ERROR(CheckMaxSupportedOpVersion(registration, 2)); RETURN_IF_ERROR(CheckInputsOutputs(context, tflite_node, /*runtime_inputs=*/2, /*outputs=*/1)); return absl::OkStatus(); @@ -2290,42 +2291,6 @@ class TransformLandmarksOperationParser : public TFLiteOperationParser { output_value->tensor.shape = graph->FindInputs(node->id)[0]->tensor.shape; return absl::OkStatus(); } - - private: -}; - -class TransformLandmarksV2OperationParser : public TFLiteOperationParser { - public: - absl::Status IsSupported(const TfLiteContext* context, - const TfLiteNode* tflite_node, - const TfLiteRegistration* registration) final { - RETURN_IF_ERROR(CheckInputsOutputs(context, tflite_node, - /*runtime_inputs=*/2, /*outputs=*/1)); - return absl::OkStatus(); - } - - absl::Status Parse(const TfLiteNode* tflite_node, - const TfLiteRegistration* registration, - GraphFloat32* graph, ObjectReader* reader) final { - Node* node = graph->NewNode(); - RETURN_IF_ERROR(reader->AddInput(node, 0)); // data - RETURN_IF_ERROR(reader->AddInput(node, 1)); // bbox - RETURN_IF_ERROR(reader->AddOutputs(node)); - std::string op_name = "transform_landmarks_v2"; - node->operation.type = op_name; - - auto output_value = graph->FindOutputs(node->id)[0]; - output_value->tensor.shape = graph->FindInputs(node->id)[0]->tensor.shape; - BHWC output_shape = output_value->tensor.shape; - RETURN_IF_ERROR(ParseCustomAttributes( - op_name, registration->version, tflite_node->custom_initial_data, - tflite_node->custom_initial_data_size, &(node->operation.attributes), - &output_shape)); - - return absl::OkStatus(); - } - - private: }; class Landmarks2TransformMatrixOperationParser : public TFLiteOperationParser { @@ -2596,9 +2561,6 @@ std::unique_ptr NewOperationParser( if (custom_name == "TransformLandmarks") { return std::make_unique(); } - if (custom_name == "TransformLandmarksV2") { - return std::make_unique(); - } if (custom_name == "Landmarks2TransformMatrix" || custom_name == "Landmarks2TransformMatrixV2") { return std::make_unique();