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

Enable ROI v2 ops: TransformTensorBilinear, TransformLandmarks, ROI2TransformMatrix.

Browse Source 
PiperOrigin-RevId: 313470232
Change-Id: Id581304ec9313070369c68bffe2fa12690e45c0b
This commit is contained in:
A. Unique TensorFlower 2020-05-27 14:58:34 -07:00 committed by TensorFlower Gardener
parent 3bb7d689d3
commit 43862c55a3
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

@ -2453,15 +2453,15 @@ class TransformLandmarksV2OperationParser : public TFLiteOperationParser {
RETURN_IF_ERROR(reader->AddOutputs(node));
std::string op_name = "transform_landmarks_v2";
node->operation.type = op_name;
BHWC output_shape;
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, tflite_node->custom_initial_data,
tflite_node->custom_initial_data_size,
&(node->operation.attributes), &output_shape));
auto output_value = graph->FindOutputs(node->id)[0];
output_value->tensor.shape = graph->FindInputs(node->id)[0]->tensor.shape;
return absl::OkStatus();
}