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

Rename TransformTensorV2 op to TransformTensorBilinearV2 op.

Browse Source 
PiperOrigin-RevId: 312184091
Change-Id: I5450142e1022f72705bc5fbdf6c99c94cdbb346b
This commit is contained in:
A. Unique TensorFlower 2020-05-18 17:09:11 -07:00 committed by TensorFlower Gardener
parent ad6798a2f6
commit acaaab2504
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -2350,7 +2350,7 @@ class TransformTensorOperationParser : public TFLiteOperationParser {
private: private:
}; };
class TransformTensorV2OperationParser : public TFLiteOperationParser { class TransformTensorBilinearV2OperationParser : public TFLiteOperationParser {
public: public:
absl::Status IsSupported(const TfLiteContext* context, absl::Status IsSupported(const TfLiteContext* context,
const TfLiteNode* tflite_node, const TfLiteNode* tflite_node,
@ -2368,7 +2368,7 @@ class TransformTensorV2OperationParser : public TFLiteOperationParser {
RETURN_IF_ERROR(reader->AddInput(node, 1)); // bbox RETURN_IF_ERROR(reader->AddInput(node, 1)); // bbox
RETURN_IF_ERROR(reader->AddOutputs(node)); 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; node->operation.type = op_name;
BHWC output_shape; BHWC output_shape;
RETURN_IF_ERROR( RETURN_IF_ERROR(
@ -2731,8 +2731,8 @@ std::unique_ptr<TFLiteOperationParser> NewOperationParser(
if (custom_name == "TransformTensor") { if (custom_name == "TransformTensor") {
return std::make_unique<TransformTensorOperationParser>(); return std::make_unique<TransformTensorOperationParser>();
} }
if (custom_name == "TransformTensorV2") { if (custom_name == "TransformTensorBilinearV2") {
return std::make_unique<TransformTensorV2OperationParser>(); return std::make_unique<TransformTensorBilinearV2OperationParser>();
} }
if (custom_name == "TransformLandmarks") { if (custom_name == "TransformLandmarks") {
return std::make_unique<TransformLandmarksOperationParser>(); return std::make_unique<TransformLandmarksOperationParser>();