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

Remove duplicate error check in RESHAPE.

Browse Source 
PiperOrigin-RevId: 296095268
Change-Id: I22c50cf2fb8cb1343c8c99f9cf557aad45aff15c
This commit is contained in:
Juhyun Lee 2020-02-19 17:25:42 -08:00 committed by TensorFlower Gardener
parent 7ad1eb110f
commit dea81b04c3
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/lite/delegates/gpu/gl/kernels/reshape.cc
View File

@ -38,14 +38,11 @@ class Reshape : public NodeShader {
auto output = ctx.graph->FindOutputs(ctx.node->id)[0]; auto output = ctx.graph->FindOutputs(ctx.node->id)[0];
if (input->tensor.shape.DimensionsProduct() != if (input->tensor.shape.DimensionsProduct() !=
output->tensor.shape.DimensionsProduct()) { output->tensor.shape.DimensionsProduct()) {
return InvalidArgumentError("Dimensions product is reshape don't match"); return InvalidArgumentError(
"Number of elements in input & output tensors don't match.");
} }
auto attr = auto attr =
absl::any_cast<ReshapeAttributes>(ctx.node->operation.attributes); absl::any_cast<ReshapeAttributes>(ctx.node->operation.attributes);
if (input->tensor.shape.DimensionsProduct() !=
output->tensor.shape.DimensionsProduct()) {
return InvalidArgumentError("Dimensions product is reshape don't match");
}
if (attr.new_shape != output->tensor.shape) { if (attr.new_shape != output->tensor.shape) {
return InvalidArgumentError( return InvalidArgumentError(
"Dimensions for output does not match new_shape attribute"); "Dimensions for output does not match new_shape attribute");