From acaaab2504a94711a4c1084328c79c10b7c9a594 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Mon, 18 May 2020 17:09:11 -0700 Subject: [PATCH] Rename TransformTensorV2 op to TransformTensorBilinearV2 op. PiperOrigin-RevId: 312184091 Change-Id: I5450142e1022f72705bc5fbdf6c99c94cdbb346b --- tensorflow/lite/delegates/gpu/common/model_builder.cc | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tensorflow/lite/delegates/gpu/common/model_builder.cc b/tensorflow/lite/delegates/gpu/common/model_builder.cc index 46856a70a7c..964c8289f83 100644 --- a/tensorflow/lite/delegates/gpu/common/model_builder.cc +++ b/tensorflow/lite/delegates/gpu/common/model_builder.cc @@ -2350,7 +2350,7 @@ class TransformTensorOperationParser : public TFLiteOperationParser { private: }; -class TransformTensorV2OperationParser : public TFLiteOperationParser { +class TransformTensorBilinearV2OperationParser : public TFLiteOperationParser { public: absl::Status IsSupported(const TfLiteContext* context, const TfLiteNode* tflite_node, @@ -2368,7 +2368,7 @@ class TransformTensorV2OperationParser : public TFLiteOperationParser { RETURN_IF_ERROR(reader->AddInput(node, 1)); // bbox RETURN_IF_ERROR(reader->AddOutputs(node)); - std::string op_name = "transform_tensor_v2"; + std::string op_name = "transform_tensor_bilinear_v2"; node->operation.type = op_name; BHWC output_shape; RETURN_IF_ERROR( @@ -2731,8 +2731,8 @@ std::unique_ptr NewOperationParser( if (custom_name == "TransformTensor") { return std::make_unique(); } - if (custom_name == "TransformTensorV2") { - return std::make_unique(); + if (custom_name == "TransformTensorBilinearV2") { + return std::make_unique(); } if (custom_name == "TransformLandmarks") { return std::make_unique();