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

Added new way of scalar reading in ParseMultiplyScalar.

Browse Source 
PiperOrigin-RevId: 316903891
Change-Id: I6a160c0a1a83cff9ee6b1df9c3f59ea0477d6c30
This commit is contained in:
Raman Sarokin 2020-06-17 09:30:39 -07:00 committed by TensorFlower Gardener
parent fa60748329
commit 225bdf60f3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1271,7 +1271,7 @@ class MulOperationParser : public TFLiteOperationParser {
GraphFloat32* graph, ObjectReader* reader) {
RETURN_IF_ERROR(reader->AddInput(node, runtime_tensor));
MultiplyAttributes attr;
if (constant_dims->size <= 0) {
if (constant_dims->size <= 0 || NumElements(constant_dims) == 1) {
Tensor<Scalar, DataType::FLOAT32> tensor;
RETURN_IF_ERROR(reader->ReadTensor(constant_tensor, &tensor));
attr.param = tensor.data[0];